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 *src1, int *src2, int *src3, int imm, int *r) 14{ 15 int i, j, index, res, mask, one_mask = 1; 16 int src1_bit, src2_bit, src3_bit, imm_bit; 17 18 for (i = 0; i < SIZE; i++) 19 { 20 res = 0; 21 for (j = 0; j < 32; j++) 22 { 23 mask = one_mask << j; 24 src1_bit = ((src1[i] & mask) >> j) << 2; 25 src2_bit = ((src2[i] & mask) >> j) << 1; 26 src3_bit = ((src3[i] & mask) >> j); 27 index = src1_bit | src2_bit | src3_bit; 28 imm_bit = (imm & (one_mask << index)) >> index; 29 res = res | (imm_bit << j); 30 } 31 r[i] = res; 32 } 33} 34 35void 36TEST (void) 37{ 38 UNION_TYPE (AVX512F_LEN, i_d) src2, src3, res1, res2, res3; 39 MASK_TYPE mask = MASK_VALUE; 40 int res_ref[SIZE]; 41 int i, imm = 0x7D; 42 43 for (i = 0; i < SIZE; i++) 44 { 45 res1.a[i] = DEFAULT_VALUE; 46 res2.a[i] = DEFAULT_VALUE; 47 res3.a[i] = DEFAULT_VALUE; 48 src2.a[i] = 145132 * i + 123123; 49 src3.a[i] = 1223 * i + 895; 50 } 51 52 CALC (res1.a, src2.a, src3.a, imm, res_ref); 53 54 res1.x = INTRINSIC (_ternarylogic_epi32) (res1.x, src2.x, src3.x, 55 imm); 56 res2.x = INTRINSIC (_mask_ternarylogic_epi32) (res2.x, mask, src2.x, 57 src3.x, imm); 58 res3.x = INTRINSIC (_maskz_ternarylogic_epi32) (mask, res3.x, src2.x, 59 src3.x, imm); 60 61 if (UNION_CHECK (AVX512F_LEN, i_d) (res1, res_ref)) 62 abort (); 63 64 MASK_MERGE (i_d) (res_ref, mask, SIZE); 65 if (UNION_CHECK (AVX512F_LEN, i_d) (res2, res_ref)) 66 abort (); 67 68 MASK_ZERO (i_d) (res_ref, mask, SIZE); 69 if (UNION_CHECK (AVX512F_LEN, i_d) (res3, res_ref)) 70 abort (); 71} 72