Viewing file:  avx512fp16-reduce-op-1.c (3.3 KB) -rw-r--r--
Select action/file-type:
 (+) |  (+) |  (+) | Code (+) | Session (+) |  (+) | SDB (+) |  (+) |  (+) |  (+) |  (+) |  (+) |
/* { dg-do run { target avx512fp16 } } */
/* { dg-options "-O2 -mavx512fp16" } */
static void do_test (void);
#define DO_TEST do_test
#define AVX512FP16
#include <immintrin.h>
#include "avx512-check.h"
__m512h a1 = { -39.3f16, -180.9f16, 13.4f16, 35.4f16, -41.1f16, -14.4f16, 24.5f16, 53.54f16,
238.4f16, -134.8f16, 24.5f16, 35.6f16, -346.7f16, -43.4f16, -535.3f16, 324.7f16,
82.5f16, 21.4f16, 24.4f16, 53.4f16, 23.5f16, -24.4f16, -34.5f16, -32.5f16,
23.6f16, -13.4f16, 24.5f16, 35.5f16, -34.4f16, -24.5f16, -34.5f16, 13.5f16 };
__m512h a2 = { 1.25f16, 2.25f16, -0.25f16, 4.0f16, -2.0f16, 4.0f16, -3.0f16, 2.0f16,
-0.5f16, -1.0f16, 1.0f16, -1.0f16, 1.0f16, 1.0f16, 2.0f16, 4.0f16,
1.25f16, 2.25f16, -4.25f16, 4.0f16, -2.4f16, 4.0f16, -3.0f, 2.0f16,
-4.5f16, 7.6f16, 0.7f16, -8.2f16, 2.1f16, 2.4f16, -2.0f16, 19.4f16 };
__attribute__((noinline, noclone)) _Float16
test_reduce_add_ph (__m512h a)
{
return _mm512_reduce_add_ph (a);
}
__attribute__((noinline, noclone)) _Float16
test_reduce_mul_ph (__m512h a)
{
return _mm512_reduce_mul_ph (a);
}
__attribute__((noinline, noclone)) _Float16
test_reduce_max_ph (__m512h a)
{
return _mm512_reduce_max_ph (a);
}
__attribute__((noinline, noclone)) _Float16
test_reduce_min_ph (__m512h a)
{
return _mm512_reduce_min_ph (a);
}
#define SIZE 32
#define REF_ADDMUL(op, a) \
__m256h __a1 = _mm256_setzero_ph (); \
for (int i =0; i < 16; i++) { \
__a1[i] = (_Float16) a[i] op (_Float16) a[i + 16]; \
} \
__m128h __a2 = _mm_setzero_ph (); \
for (int i =0; i < 8; i++) { \
__a2[i] = (_Float16) __a1[i] op (_Float16) __a1[i + 8]; \
} \
_Float16 __c0 = __a2[0] op __a2[4]; \
_Float16 __c1 = __a2[1] op __a2[5]; \
_Float16 __c2 = __a2[2] op __a2[6]; \
_Float16 __c3 = __a2[3] op __a2[7]; \
_Float16 __d0 = __c0 op __c2; \
_Float16 __d1 = __c1 op __c3; \
_Float16 __e0 = __d0 op __d1; \
r3 = __e0
#define TESTOP(opname, op, a) \
do { \
_Float16 r1 = _mm512_reduce_##opname##_ph (a); \
_Float16 r2 = test_reduce_##opname##_ph (a); \
_Float16 r3 = a[0]; \
if (r1 != r2) { \
__builtin_abort (); \
} \
REF_ADDMUL (op, a); \
if (r1 != r3) { \
__builtin_abort (); \
} \
} while (0)
#define TEST_ADDMUL_PH(a) \
do { \
TESTOP (add, +, a); \
TESTOP (mul, *, a); \
} while (0)
static void
test_512_addmul_ph (void)
{
TEST_ADDMUL_PH (a1);
TEST_ADDMUL_PH (a2);
}
#undef TESTOP
#define TESTOP(opname, op, a) \
do { \
_Float16 r1 = _mm512_reduce_##opname##_ph (a); \
_Float16 r2 = test_reduce_##opname##_ph (a); \
_Float16 r3 = a[0]; \
if (r1 != r2) { \
__builtin_abort (); \
} \
for (int i = 1; i < SIZE; i++) \
r3 = r3 op a[i]; \
if (r1 != r3) { \
__builtin_abort (); \
} \
} while (0)
#define TEST_MINMAX_PH(a) \
do { \
TESTOP (min, < a[i] ? r3 :, a); \
TESTOP (max, > a[i] ? r3 :, a); \
} while (0)
static void
test_512_minmax_ph (void)
{
TEST_MINMAX_PH (a1);
TEST_MINMAX_PH (a2);
}
static void
do_test (void)
{
test_512_addmul_ph();
test_512_minmax_ph();
}
#undef SIZE
#undef REF_ADDMUL
#undef TESTOP
#undef TEST_ADDMUL_PH
#undef TEST_MINMAX_PH
|
