Viewing file:  bf16_vldN_lane_1.c (2.37 KB) -rw-r--r--
Select action/file-type:
 (+) |  (+) |  (+) | Code (+) | Session (+) |  (+) | SDB (+) |  (+) |  (+) |  (+) |  (+) |  (+) |
/* { dg-do run { target { aarch64*-*-* } } } */
/* { dg-require-effective-target arm_v8_2a_bf16_neon_ok } */
/* { dg-add-options arm_v8_2a_bf16_neon } */
#include <arm_neon.h>
extern void abort (void);
typedef union
{
bfloat16_t bf16;
uint16_t u16;
} bfloat16_u_t;
#define VARIANTS(VARIANT, STRUCT) \
VARIANT (bfloat16, , 4, _bf16, 3, STRUCT) \
VARIANT (bfloat16, q, 8, _bf16, 7, STRUCT)
#define TESTMETH(BASE, Q, ELTS, SUFFIX, LANE, STRUCT) \
int \
test_vld##STRUCT##Q##_lane##SUFFIX (const bfloat16_u_t *data, \
const bfloat16_u_t *overwrite) \
{ \
BASE##x##ELTS##x##STRUCT##_t vectors; \
bfloat16_u_t temp[ELTS]; \
int i,j; \
for (i = 0; i < STRUCT; i++, data += ELTS) \
vectors.val[i] = vld1##Q##SUFFIX ((bfloat16_t *)data); \
vectors = vld##STRUCT##Q##_lane##SUFFIX ((bfloat16_t *) overwrite, \
vectors, LANE); \
while (--i >= 0) \
{ \
vst1##Q##SUFFIX ((bfloat16_t *)temp, vectors.val[i]); \
data -= ELTS; /* Point at value loaded before vldN_lane. */ \
for (j = 0; j < ELTS; j++) \
if (temp[j].u16 != (j == LANE ? overwrite[i].u16 : data[j].u16)) \
return 1; \
} \
return 0; \
}
/* Tests of vld2_lane and vld2q_lane. */
VARIANTS (TESTMETH, 2)
/* Tests of vld3_lane and vld3q_lane. */
VARIANTS (TESTMETH, 3)
/* Tests of vld4_lane and vld4q_lane. */
VARIANTS (TESTMETH, 4)
#define CHECK(BASE, Q, ELTS, SUFFIX, LANE, STRUCT) \
if (test_vld##STRUCT##Q##_lane##SUFFIX ((const bfloat16_u_t *)orig_data, \
BASE##_data) != 0) \
abort ();
int
main (int argc, char **argv)
{
/* Original data for all vector formats. */
uint64_t orig_data[8] = {0x1234567890abcdefULL, 0x13579bdf02468aceULL,
0x012389ab4567cdefULL, 0xdeeddadacafe0431ULL,
0x1032547698badcfeULL, 0xbadbadbadbad0badULL,
0x0102030405060708ULL, 0x0f0e0d0c0b0a0908ULL};
/* Data with which vldN_lane will overwrite some of previous. */
bfloat16_u_t bfloat16_data[4];
bfloat16_data[0].u16 = 0xABAB;
bfloat16_data[1].u16 = 0x0;
bfloat16_data[2].u16 = 0xCAFE;
bfloat16_data[3].u16 = 0x1234;
VARIANTS (CHECK, 2);
VARIANTS (CHECK, 3);
VARIANTS (CHECK, 4);
return 0;
}
|
