[ENDIAN] Endian testing (#253)

dmlc · May 17, 2017 · d100d0f · d100d0f
1 parent e6e9e80
commit d100d0f
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/include/dmlc/endian.h b/include/dmlc/endian.h
@@ -0,0 +1,17 @@
+/*!
+ *  Copyright (c) 2017 by Contributors
+ * \file endian.h
+ * \brief Endian testing
+ */
+#ifndef DMLC_ENDIAN_H_
+#define DMLC_ENDIAN_H_
+
+#if defined(__APPLE__) || defined(_WIN32)
+#define DMLC_LITTLE_ENDIAN 1
+#else
+#include <endian.h>
+#define DMLC_LITTLE_ENDIAN (__BYTE_ORDER == __LITTLE_ENDIAN)
+#endif
+
+#endif  // DMLC_ENDIAN_H_
+
diff --git a/test/filesys_test.cc b/test/filesys_test.cc
@@ -2,6 +2,7 @@
 #include <cstdlib>
 #include <dmlc/logging.h>
 #include <dmlc/io.h>
+#include <dmlc/endian.h>
 #include "../src/io/filesys.h"
 
 int main(int argc, char *argv[]) {