1/* { dg-do run } */
2/* { dg-require-effective-target aes } */
3/* { dg-options "-O2 -maes" } */
4
5#ifndef CHECK_H
6#define CHECK_H "aes-check.h"
7#endif
8
9#ifndef TEST
10#define TEST aes_test
11#endif
12
13#include CHECK_H
14
15#include <wmmintrin.h>
16#include <string.h>
17
18extern void abort (void);
19
20#define NUM 1024
21#define IMM8 1
22
23static __m128i src1[NUM];
24static __m128i edst[NUM];
25
26static __m128i resdst[NUM];
27
28/* Initialize input/output vectors.  (Currently, there is only one set
29   of input/output vectors).  */
30
31static void
32init_data (__m128i *s1, __m128i *d)
33{
34  int i;
35  for (i = 0; i < NUM; i++)
36    {
37      s1[i] = _mm_setr_epi32 (0x16157e2b, 0xa6d2ae28,
38			      0x8815f7ab, 0x3c4fcf09);
39      d[i] = _mm_setr_epi32 (0x24b5e434, 0x3424b5e5,
40			     0xeb848a01, 0x01eb848b);
41    }
42}
43
44static void
45TEST (void)
46{
47  int i;
48
49  init_data (src1, edst);
50
51  for (i = 0; i < NUM; i += 16)
52    {
53      resdst[i]  = _mm_aeskeygenassist_si128 (src1[i], IMM8);
54      resdst[i + 1] = _mm_aeskeygenassist_si128 (src1[i + 1], IMM8);
55      resdst[i + 2] = _mm_aeskeygenassist_si128 (src1[i + 2], IMM8);
56      resdst[i + 3] = _mm_aeskeygenassist_si128 (src1[i + 3], IMM8);
57      resdst[i + 4] = _mm_aeskeygenassist_si128 (src1[i + 4], IMM8);
58      resdst[i + 5] = _mm_aeskeygenassist_si128 (src1[i + 5], IMM8);
59      resdst[i + 6] = _mm_aeskeygenassist_si128 (src1[i + 6], IMM8);
60      resdst[i + 7] = _mm_aeskeygenassist_si128 (src1[i + 7], IMM8);
61      resdst[i + 8] = _mm_aeskeygenassist_si128 (src1[i + 8], IMM8);
62      resdst[i + 9] = _mm_aeskeygenassist_si128 (src1[i + 9], IMM8);
63      resdst[i + 10] = _mm_aeskeygenassist_si128 (src1[i + 10], IMM8);
64      resdst[i + 11] = _mm_aeskeygenassist_si128 (src1[i + 11], IMM8);
65      resdst[i + 12] = _mm_aeskeygenassist_si128 (src1[i + 12], IMM8);
66      resdst[i + 13] = _mm_aeskeygenassist_si128 (src1[i + 13], IMM8);
67      resdst[i + 14] = _mm_aeskeygenassist_si128 (src1[i + 14], IMM8);
68      resdst[i + 15] = _mm_aeskeygenassist_si128 (src1[i + 15], IMM8);
69    }
70
71  for (i = 0; i < NUM; i++)
72    if (memcmp(edst + i, resdst + i, sizeof (__m128i)))
73      abort ();
74}
75