1/* { dg-do run } */ 2/* { dg-options "-O2 -mavx512f" } */ 3/* { dg-require-effective-target avx512f } */ 4 5#define AVX512F 6 7#include "avx512f-helper.h" 8 9#define SIZE (AVX512F_LEN / 32) 10#include "avx512f-mask-type.h" 11 12static void 13CALC (int *r, int *s1, int count) 14{ 15 int i; 16 for (i = 0; i < SIZE; i++) 17 { 18 r[i] = count < 32 ? (s1[i] << count) : 0; 19 } 20} 21 22void 23TEST (void) 24{ 25 int i, sign; 26 UNION_TYPE (AVX512F_LEN, i_d) res1, res2, res3, src1; 27 MASK_TYPE mask = MASK_VALUE; 28 int res_ref[SIZE]; 29 30 sign = -1; 31 for (i = 0; i < SIZE; i++) 32 { 33 src1.a[i] = 2 + sign * 7 * i % 291; 34 sign = sign * -1; 35 } 36 37 for (i = 0; i < SIZE; i++) 38 res2.a[i] = DEFAULT_VALUE; 39 40 res1.x = INTRINSIC (_slli_epi32) (src1.x, 2); 41 res2.x = INTRINSIC (_mask_slli_epi32) (res2.x, mask, src1.x, 2); 42 res3.x = INTRINSIC (_maskz_slli_epi32) (mask, src1.x, 2); 43 44 CALC (res_ref, src1.a, 2); 45 46 if (UNION_CHECK (AVX512F_LEN, i_d) (res1, res_ref)) 47 abort (); 48 49 MASK_MERGE (i_d) (res_ref, mask, SIZE); 50 if (UNION_CHECK (AVX512F_LEN, i_d) (res2, res_ref)) 51 abort (); 52 53 MASK_ZERO (i_d) (res_ref, mask, SIZE); 54 if (UNION_CHECK (AVX512F_LEN, i_d) (res3, res_ref)) 55 abort (); 56 57 58 for (i = 0; i < SIZE; i++) 59 res2.a[i] = DEFAULT_VALUE; 60 61 res1.x = INTRINSIC (_slli_epi32) (src1.x, 33); 62 res2.x = INTRINSIC (_mask_slli_epi32) (res2.x, mask, src1.x, 33); 63 res3.x = INTRINSIC (_maskz_slli_epi32) (mask, src1.x, 33); 64 65 CALC (res_ref, src1.a, 33); 66 67 if (UNION_CHECK (AVX512F_LEN, i_d) (res1, res_ref)) 68 abort (); 69 70 MASK_MERGE (i_d) (res_ref, mask, SIZE); 71 if (UNION_CHECK (AVX512F_LEN, i_d) (res2, res_ref)) 72 abort (); 73 74 MASK_ZERO (i_d) (res_ref, mask, SIZE); 75 if (UNION_CHECK (AVX512F_LEN, i_d) (res3, res_ref)) 76 abort (); 77} 78