From 8c87484fc0bf8a2ae68c497264206c95998eb088 Mon Sep 17 00:00:00 2001
From: Sidong-Wei <Sidong.Wei@ibm.com>
Date: Thu, 19 Nov 2020 11:54:12 -0500
Subject: [PATCH] Add fix for lsh_projection_test
---
tensorflow/lite/kernels/lsh_projection_test.cc | 15 +++++++++++++++
1 file changed, 15 insertions(+)
diff --git a/tensorflow/lite/kernels/lsh_projection_test.cc b/tensorflow/lite/kernels/lsh_projection_test.cc
index 008a5c45aaa..b05ea10a195 100644
--- a/tensorflow/lite/kernels/lsh_projection_test.cc
+++ b/tensorflow/lite/kernels/lsh_projection_test.cc
@@ -87,7 +87,12 @@ TEST(LSHProjectionOpTest2, Dense1DInputs) {
m.Invoke();
+#if defined(__s390x__)
+ // Hash returns differently on machines with different endianness
+ EXPECT_THAT(m.GetOutput(), ElementsAre(0, 0, 1, 1, 1, 0));
+#else
EXPECT_THAT(m.GetOutput(), ElementsAre(0, 0, 0, 1, 0, 0));
+#endif
}
TEST(LSHProjectionOpTest2, Sparse1DInputs) {
@@ -98,7 +103,12 @@ TEST(LSHProjectionOpTest2, Sparse1DInputs) {
m.Invoke();
+#if defined(__s390x__)
+ // Hash returns differently on machines with different endianness
+ EXPECT_THAT(m.GetOutput(), ElementsAre(0 + 0, 4 + 3, 8 + 2));
+#else
EXPECT_THAT(m.GetOutput(), ElementsAre(0 + 0, 4 + 1, 8 + 0));
+#endif
}
TEST(LSHProjectionOpTest2, Sparse3DInputs) {
@@ -111,7 +121,12 @@ TEST(LSHProjectionOpTest2, Sparse3DInputs) {
m.Invoke();
+#if defined(__s390x__)
+ // Hash returns differently on machines with different endianness
+ EXPECT_THAT(m.GetOutput(), ElementsAre(0 + 0, 4 + 3, 8 + 2));
+#else
EXPECT_THAT(m.GetOutput(), ElementsAre(0 + 2, 4 + 1, 8 + 1));
+#endif
}
} // namespace