diff options
author | Rasmus Munk Larsen <rmlarsen@google.com> | 2020-10-09 20:05:49 +0000 |
---|---|---|
committer | Rasmus Munk Larsen <rmlarsen@google.com> | 2020-10-09 20:05:49 +0000 |
commit | 4e4d3f32d168ed9ce09d950f099a60ddcd11240f (patch) | |
tree | 3e52ae5b43c238679f69f3caf4d908d4afb16f13 /test/packetmath_test_shared.h | |
parent | 7a8d3d5b81cb528f7f084b63686ffb20494053f6 (diff) |
Clean up packetmath tests and fix various bugs to make bfloat16 pass (almost) all packetmath tests with SSE, AVX, and AVX512.
Diffstat (limited to 'test/packetmath_test_shared.h')
-rw-r--r-- | test/packetmath_test_shared.h | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/test/packetmath_test_shared.h b/test/packetmath_test_shared.h index 7b8caedcb..f8dc3711c 100644 --- a/test/packetmath_test_shared.h +++ b/test/packetmath_test_shared.h @@ -156,7 +156,7 @@ struct packet_helper<false,Packet> #define CHECK_CWISE1_IF(COND, REFOP, POP) if(COND) { \ test::packet_helper<COND,Packet> h; \ for (int i=0; i<PacketSize; ++i) \ - ref[i] = REFOP(data1[i]); \ + ref[i] = Scalar(REFOP(data1[i])); \ h.store(data2, POP(h.load(data1))); \ VERIFY(test::areApprox(ref, data2, PacketSize) && #POP); \ } @@ -164,7 +164,7 @@ struct packet_helper<false,Packet> #define CHECK_CWISE2_IF(COND, REFOP, POP) if(COND) { \ test::packet_helper<COND,Packet> h; \ for (int i=0; i<PacketSize; ++i) \ - ref[i] = REFOP(data1[i], data1[i+PacketSize]); \ + ref[i] = Scalar(REFOP(data1[i], data1[i+PacketSize])); \ h.store(data2, POP(h.load(data1),h.load(data1+PacketSize))); \ VERIFY(test::areApprox(ref, data2, PacketSize) && #POP); \ } @@ -172,8 +172,8 @@ struct packet_helper<false,Packet> #define CHECK_CWISE3_IF(COND, REFOP, POP) if (COND) { \ test::packet_helper<COND, Packet> h; \ for (int i = 0; i < PacketSize; ++i) \ - ref[i] = \ - REFOP(data1[i], data1[i + PacketSize], data1[i + 2 * PacketSize]); \ + ref[i] = Scalar(REFOP(data1[i], data1[i + PacketSize], \ + data1[i + 2 * PacketSize])); \ h.store(data2, POP(h.load(data1), h.load(data1 + PacketSize), \ h.load(data1 + 2 * PacketSize))); \ VERIFY(test::areApprox(ref, data2, PacketSize) && #POP); \ |