fix arm cmake (#93)

bytedance · Aug 28, 2024 · 097b3a0 · 097b3a0
1 parent 2fc19d4
commit 097b3a0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/cmake/set_arch_flags.cmake b/cmake/set_arch_flags.cmake
@@ -2,7 +2,7 @@ function(set_arch_flags target arch)
   message(STATUS "Setting architecture flags for ${arch}")
   if(arch MATCHES "x86_64")
     target_compile_options(${target} PRIVATE -mavx2 -mpclmul -mbmi -mlzcnt)
-  elseif(arch MATCHES "arm")
+  elseif(arch MATCHES "aarch64")
     target_compile_options(${target} PRIVATE -march=armv8-a)
   else()
     message(FATAL_ERROR "Unsupported architecture: ${arch}")

diff --git a/scripts/unittest.sh b/scripts/unittest.sh
@@ -9,7 +9,7 @@ Usage:
     -g, --gcc       compiler is gcc
     -c, --clang     compiler is clang
     -h, --help      display this message
-    --arch={arm|haswell|westmere} target architecture, default is haswell
+    --arch={aarch64|haswell|westmere} target architecture, default is haswell
     --dispatch={dynamic|static} sonic dispatch mode, default is static
 
     example: bash unittest.sh -g --arch=westmere --dispatch=static
@@ -43,7 +43,7 @@ do
     --arch)
         case "$2" in
             "") shift 2 ;;
-            arm|haswell|westmere)
+            aarch64|haswell|westmere)
                 UNIT_TEST_ARCH="$2"
                 shift 2 ;;
             *)