Add support for non-x86_64 architectures (e.g. Linux ARM64)

Fixes voutcn#320 Based on: voutcn#329 Preserves the x86_64 specifics when building on x86_64 machine Signed-off-by: Martin Tzvetanov Grigorov <[email protected]>
martin-g · Mar 27, 2024 · 296ad6b · 296ad6b
1 parent 5f329c6
commit 296ad6b
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 2 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -73,8 +73,12 @@ add_executable(megahit_core_popcnt ${OTHER_SOURCE} ${ASMBL_SOURCE} ${IDBA_SOURCE
         ${CX1_SOURCE} ${TOOLKIT_SOURCE})
 add_executable(megahit_core_no_hw_accel ${OTHER_SOURCE} ${ASMBL_SOURCE} ${IDBA_SOURCE} ${SDBG_SOURCE} ${LCASM_SOURCE}
         ${SEQ_SOURCE} ${CX1_SOURCE} ${TOOLKIT_SOURCE})
-set_target_properties(megahit_core PROPERTIES COMPILE_FLAGS "-mbmi2 -DUSE_BMI2 -mpopcnt")
-set_target_properties(megahit_core_popcnt PROPERTIES COMPILE_FLAGS "-mpopcnt")
+
+if (CMAKE_SYSTEM_PROCESSOR MATCHES "(x86_64)|(amd64)|(AMD64)")
+    set_target_properties(megahit_core PROPERTIES COMPILE_FLAGS "-mbmi2 -DUSE_BMI2 -mpopcnt")
+    set_target_properties(megahit_core_popcnt PROPERTIES COMPILE_FLAGS "-mpopcnt")
+endif ()
+
 
 if (STATIC_BUILD)
     # TODO dirty

diff --git a/src/kmlib/kmrns.h b/src/kmlib/kmrns.h
@@ -8,7 +8,9 @@
 #include <stdint.h>
 #include <stdio.h>
 #include <stdlib.h>
+#if defined(__x86_64__)
 #include <x86intrin.h>
+#endif
 #include <vector>
 
 namespace kmlib {

diff --git a/src/utils/cpu_dispatch.h b/src/utils/cpu_dispatch.h
@@ -5,6 +5,7 @@
 #ifndef MEGAHIT_CPU_DISPATCH_H
 #define MEGAHIT_CPU_DISPATCH_H
 
+#if defined(__x86_64__)
 inline bool HasPopcnt() {
   unsigned eax, ebx, ecx, edx;
 #ifdef _MSC_VER
@@ -32,5 +33,10 @@ inline bool HasBmi2() {
 #endif
   return ebx >> 8U & 1U;
 }
+#else
+inline bool HasPopcnt() { return false; }
+inline bool HasBmi2() { return false; }
+#endif
+
 
 #endif  // MEGAHIT_CPU_DISPATCH_H