experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Make lsh_projection_test fits all big endian systems

Browse Source
This commit is contained in:
Sidong-Wei 2020-11-26 08:14:21 -05:00
parent 78d22273ad
commit 05f2326d01
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tensorflow/lite/kernels/lsh_projection_test.cc
View File

@ -87,7 +87,8 @@ TEST(LSHProjectionOpTest2, Dense1DInputs) {
m.Invoke(); m.Invoke();
#if defined(__s390x__) #if defined(__BYTE_ORDER__) && defined(__ORDER_BIG_ENDIAN__) && \
__BYTE_ORDER__ == __ORDER_BIG_ENDIAN__
// Hash returns differently on machines with different endianness // Hash returns differently on machines with different endianness
EXPECT_THAT(m.GetOutput(), ElementsAre(0, 0, 1, 1, 1, 0)); EXPECT_THAT(m.GetOutput(), ElementsAre(0, 0, 1, 1, 1, 0));
#else #else
@ -103,7 +104,8 @@ TEST(LSHProjectionOpTest2, Sparse1DInputs) {
m.Invoke(); m.Invoke();
#if defined(__s390x__) #if defined(__BYTE_ORDER__) && defined(__ORDER_BIG_ENDIAN__) && \
__BYTE_ORDER__ == __ORDER_BIG_ENDIAN__
// Hash returns differently on machines with different endianness // Hash returns differently on machines with different endianness
EXPECT_THAT(m.GetOutput(), ElementsAre(0 + 0, 4 + 3, 8 + 2)); EXPECT_THAT(m.GetOutput(), ElementsAre(0 + 0, 4 + 3, 8 + 2));
#else #else
@ -121,7 +123,8 @@ TEST(LSHProjectionOpTest2, Sparse3DInputs) {
m.Invoke(); m.Invoke();
#if defined(__s390x__) #if defined(__BYTE_ORDER__) && defined(__ORDER_BIG_ENDIAN__) && \
__BYTE_ORDER__ == __ORDER_BIG_ENDIAN__
// Hash returns differently on machines with different endianness // Hash returns differently on machines with different endianness
EXPECT_THAT(m.GetOutput(), ElementsAre(0 + 0, 4 + 3, 8 + 2)); EXPECT_THAT(m.GetOutput(), ElementsAre(0 + 0, 4 + 3, 8 + 2));
#else #else