Merge pull request #617 from mhsmith/master

Add missing #include for BYTE_ORDER
google · Jan 12, 2021 · ba7e11a · ba7e11a
2 parents d9a0892 + 4bc9ae0
commit ba7e11a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/util.h b/src/util.h
@@ -36,9 +36,12 @@
 #include <pthread.h>
 #endif
 
-#if !defined(__APPLE__) && !defined(_WIN32) && BYTE_ORDER == __BIG_ENDIAN
+#if !defined(__APPLE__) && !defined(_WIN32)
+#include <endian.h>
+#if defined(BYTE_ORDER) && defined(__BIG_ENDIAN) && BYTE_ORDER == __BIG_ENDIAN
 #define IS_BIG_ENDIAN
 #endif
+#endif
 
 namespace sentencepiece {
 template <typename T>