From 05f2326d01be8658da87a817b3120211ddcf7fa8 Mon Sep 17 00:00:00 2001 From: Sidong-Wei Date: Thu, 26 Nov 2020 08:14:21 -0500 Subject: [PATCH] Make lsh_projection_test fits all big endian systems --- tensorflow/lite/kernels/lsh_projection_test.cc | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/tensorflow/lite/kernels/lsh_projection_test.cc b/tensorflow/lite/kernels/lsh_projection_test.cc index b05ea10a195..a716d6c50dd 100644 --- a/tensorflow/lite/kernels/lsh_projection_test.cc +++ b/tensorflow/lite/kernels/lsh_projection_test.cc @@ -87,7 +87,8 @@ TEST(LSHProjectionOpTest2, Dense1DInputs) { 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 EXPECT_THAT(m.GetOutput(), ElementsAre(0, 0, 1, 1, 1, 0)); #else @@ -103,7 +104,8 @@ TEST(LSHProjectionOpTest2, Sparse1DInputs) { 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 EXPECT_THAT(m.GetOutput(), ElementsAre(0 + 0, 4 + 3, 8 + 2)); #else @@ -121,7 +123,8 @@ TEST(LSHProjectionOpTest2, Sparse3DInputs) { 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 EXPECT_THAT(m.GetOutput(), ElementsAre(0 + 0, 4 + 3, 8 + 2)); #else