@@ -29,8 +29,6 @@ bool check(float a, float b) {
29
29
return fabs (2 * (a - b) / (a + b)) > bf16_eps * 2 ;
30
30
}
31
31
32
- bool check (bool a, bool b) { return (a == b); }
33
-
34
32
#define TEST_BUILTIN_1_SCAL_IMPL (NAME ) \
35
33
{ \
36
34
buffer<float > a_buf (&a[0 ], N); \
@@ -48,7 +46,7 @@ bool check(bool a, bool b) { return (a == b); }
48
46
} \
49
47
assert (err == 0 );
50
48
51
- #define TEST_BUILTIN_1_ARR_IMPL (NAME, SZ, RETTY ) \
49
+ #define TEST_BUILTIN_1_ARR_IMPL (NAME, SZ ) \
52
50
{ \
53
51
buffer<float , 2 > a_buf{range<2 >{N / SZ, SZ}}; \
54
52
buffer<int > err_buf (&err, 1 ); \
@@ -61,7 +59,7 @@ bool check(bool a, bool b) { return (a == b); }
61
59
for (int i = 0 ; i < SZ; i++) { \
62
60
arg[i] = A[index][i]; \
63
61
} \
64
- marray<RETTY , SZ> res = NAME (arg); \
62
+ marray<bfloat16 , SZ> res = NAME (arg); \
65
63
for (int i = 0 ; i < SZ; i++) { \
66
64
if (check (res[i], NAME (A[index][i]))) { \
67
65
ERR[0 ] = 1 ; \
@@ -72,13 +70,13 @@ bool check(bool a, bool b) { return (a == b); }
72
70
} \
73
71
assert (err == 0 );
74
72
75
- #define TEST_BUILTIN_1 (NAME, RETTY ) \
73
+ #define TEST_BUILTIN_1 (NAME ) \
76
74
TEST_BUILTIN_1_SCAL_IMPL (NAME) \
77
- TEST_BUILTIN_1_ARR_IMPL(NAME, 1 , RETTY) \
78
- TEST_BUILTIN_1_ARR_IMPL(NAME, 2 , RETTY) \
79
- TEST_BUILTIN_1_ARR_IMPL(NAME, 3 , RETTY) \
80
- TEST_BUILTIN_1_ARR_IMPL(NAME, 4 , RETTY) \
81
- TEST_BUILTIN_1_ARR_IMPL(NAME, 5 , RETTY )
75
+ TEST_BUILTIN_1_ARR_IMPL(NAME, 1 ) \
76
+ TEST_BUILTIN_1_ARR_IMPL(NAME, 2 ) \
77
+ TEST_BUILTIN_1_ARR_IMPL(NAME, 3 ) \
78
+ TEST_BUILTIN_1_ARR_IMPL(NAME, 4 ) \
79
+ TEST_BUILTIN_1_ARR_IMPL(NAME, 5 )
82
80
83
81
#define TEST_BUILTIN_2_SCAL_IMPL (NAME ) \
84
82
{ \
@@ -235,18 +233,14 @@ int main() {
235
233
c[i] = (float )(3 * i);
236
234
}
237
235
238
- TEST_BUILTIN_1 (fabs, bfloat16 );
236
+ TEST_BUILTIN_1 (fabs);
239
237
TEST_BUILTIN_2 (fmin);
240
238
TEST_BUILTIN_2 (fmax);
241
239
TEST_BUILTIN_3 (fma);
242
240
243
241
float check_nan = 0 ;
244
242
TEST_BUILTIN_2_NAN (fmin);
245
243
TEST_BUILTIN_2_NAN (fmax);
246
-
247
- // Insert NAN value in a to test isnan
248
- a[0 ] = a[N - 1 ] = NAN;
249
- TEST_BUILTIN_1 (isnan, bool );
250
244
}
251
245
return 0 ;
252
246
}
0 commit comments