| /* { dg-do run } */ |
| /* { dg-options "-O2 -mavxvnniint8" } */ |
| /* { dg-require-effective-target avxvnniint8 } */ |
| #define AVXVNNIINT8 |
| #ifndef CHECK |
| #define CHECK "avx-check.h" |
| #endif |
| |
| #ifndef TEST |
| #define TEST avx_test |
| #endif |
| |
| #include CHECK |
| |
| static void |
| CALC (unsigned int *r, unsigned int *dst, unsigned char *s1, unsigned char *s2, int size) |
| { |
| unsigned short tempres[32]; |
| for (int i = 0; i < size; i++) { |
| tempres[i] = (unsigned short) s1[i] * (unsigned short) s2[i]; |
| } |
| for (int i = 0; i < size / 4; i++) { |
| unsigned int test = (unsigned int) dst[i] + tempres[i * 4] + tempres[i * 4 + 1] |
| + tempres[i * 4 + 2] + tempres[i * 4 + 3]; |
| r[i] = test; |
| } |
| } |
| |
| void |
| TEST (void) |
| { |
| int i; |
| union256i_ud res_256; |
| union256i_ub src2_256; |
| union256i_ub src1_256; |
| unsigned int res_ref_256[8]; |
| |
| for (i = 0; i < 32; i++) |
| { |
| int sign = i % 2 ? 1 : -1; |
| src1_256.a[i] = 10 + 3 * i + sign; |
| src2_256.a[i] = sign * 10 * i * i; |
| } |
| |
| for (i = 0; i < 8; i++) |
| res_256.a[i] = 0x7fffffff; |
| |
| CALC (res_ref_256, res_256.a, src1_256.a, src2_256.a, 32); |
| res_256.x = _mm256_dpbuud_epi32 (res_256.x, src1_256.x, src2_256.x); |
| if (check_union256i_ud (res_256, res_ref_256)) |
| abort (); |
| |
| union128i_ud res_128; |
| union128i_ub src2_128; |
| union128i_ub src1_128; |
| unsigned int res_ref_128[4]; |
| |
| for (i = 0; i < 16; i++) |
| { |
| int sign = i % 2 ? 1 : -1; |
| src1_128.a[i] = 10 + 3 * i * i + sign; |
| src2_128.a[i] = sign * 10 * i * i; |
| } |
| |
| for (i = 0; i < 4; i++) |
| res_128.a[i] = 0x7fffffff; |
| |
| CALC (res_ref_128, res_128.a, src1_128.a, src2_128.a, 16); |
| res_128.x = _mm_dpbuud_epi32 (res_128.x, src1_128.x, src2_128.x); |
| if (check_union128i_ud (res_128, res_ref_128)) |
| abort (); |
| } |