Fix: Propagate big-endian & SIMD flags

Author: ashvardanian

CMakeLists.txt

@@ -337,14 +337,14 @@ function (set_compiler_flags target cpp_standard target_arch compiler_id)
         endif ()
     endif ()
 
-    # Define SZ_DETECT_BIG_ENDIAN macro based on system byte order
+    # Define SZ_IS_BIG_ENDIAN_ macro based on system byte order
     if (CMAKE_C_BYTE_ORDER STREQUAL "BIG_ENDIAN")
-        set(SZ_DETECT_BIG_ENDIAN 1)
+        set(SZ_IS_BIG_ENDIAN_ 1)
     else ()
-        set(SZ_DETECT_BIG_ENDIAN 0)
+        set(SZ_IS_BIG_ENDIAN_ 0)
     endif ()
 
-    target_compile_definitions(${target} PRIVATE "SZ_DETECT_BIG_ENDIAN=${SZ_DETECT_BIG_ENDIAN}")
+    target_compile_definitions(${target} PRIVATE "SZ_IS_BIG_ENDIAN_=${SZ_IS_BIG_ENDIAN_}")
 
     # Sanitizer options for Debug mode
     if (CMAKE_BUILD_TYPE STREQUAL "Debug")

build.rs

@@ -41,11 +41,11 @@ fn build_stringzilla() -> HashMap<String, bool> {
 
     // Set endian-specific macro
     if target_endian == "big" {
-        build.define("SZ_DETECT_BIG_ENDIAN", "1");
-        flags.insert("SZ_DETECT_BIG_ENDIAN".to_string(), true);
+        build.define("SZ_IS_BIG_ENDIAN_", "1");
+        flags.insert("SZ_IS_BIG_ENDIAN_".to_string(), true);
     } else {
-        build.define("SZ_DETECT_BIG_ENDIAN", "0");
-        flags.insert("SZ_DETECT_BIG_ENDIAN".to_string(), false);
+        build.define("SZ_IS_BIG_ENDIAN_", "0");
+        flags.insert("SZ_IS_BIG_ENDIAN_".to_string(), false);
     }
 
     if target_arch == "x86_64" && target_bits == "64" {

include/stringzilla/types.h

@@ -218,63 +218,63 @@
  *  All of those can be controlled by the user.
  */
 #if !defined(SZ_USE_HASWELL)
-#ifdef __AVX2__
+#if SZ_IS_64BIT_X86_ && defined(__AVX2__)
 #define SZ_USE_HASWELL (1)
 #else
 #define SZ_USE_HASWELL (0)
 #endif
 #endif
 
 #if !defined(SZ_USE_SKYLAKE)
-#ifdef __AVX512F__
+#if SZ_IS_64BIT_X86_ && defined(__AVX512F__)
 #define SZ_USE_SKYLAKE (1)
 #else
 #define SZ_USE_SKYLAKE (0)
 #endif
 #endif
 
 #if !defined(SZ_USE_ICE)
-#ifdef __AVX512BW__
+#if SZ_IS_64BIT_X86_ && defined(__AVX512BW__)
 #define SZ_USE_ICE (1)
 #else
 #define SZ_USE_ICE (0)
 #endif
 #endif
 
 #if !defined(SZ_USE_NEON)
-#ifdef __ARM_NEON
+#if SZ_IS_64BIT_ARM_ && defined(__ARM_NEON)
 #define SZ_USE_NEON (1)
 #else
 #define SZ_USE_NEON (0)
 #endif
 #endif
 
 #if !defined(SZ_USE_SVE)
-#ifdef __ARM_FEATURE_SVE
+#if SZ_IS_64BIT_ARM_ && defined(__ARM_FEATURE_SVE)
 #define SZ_USE_SVE (1)
 #else
 #define SZ_USE_SVE (0)
 #endif
 #endif
 
 #if !defined(SZ_USE_SVE2)
-#ifdef __ARM_FEATURE_SVE2
+#if SZ_IS_64BIT_ARM_ && defined(__ARM_FEATURE_SVE2)
 #define SZ_USE_SVE2 (1)
 #else
 #define SZ_USE_SVE2 (0)
 #endif
 #endif
 
 #if !defined(SZ_USE_NEON_AES)
-#ifdef __ARM_FEATURE_AES
+#if SZ_IS_64BIT_ARM_ && defined(__ARM_FEATURE_AES)
 #define SZ_USE_NEON_AES (1)
 #else
 #define SZ_USE_NEON_AES (0)
 #endif
 #endif
 
 #if !defined(SZ_USE_SVE2_AES)
-#ifdef __ARM_FEATURE_SVE2_AES
+#if SZ_IS_64BIT_ARM_ && defined(__ARM_FEATURE_SVE2_AES)
 #define SZ_USE_SVE2_AES (1)
 #else
 #define SZ_USE_SVE2_AES (0)