Skip to content

Commit d6dbc14

Browse files
committed
Merge remote-tracking branch 'upstream/feat/win-compilation' into zephyr
2 parents 49a856d + 54d119b commit d6dbc14

3 files changed

Lines changed: 6 additions & 9 deletions

File tree

CMakeLists.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ message(STATUS "PROJECT_ROOT_DIR = ${PROJECT_ROOT_DIR}")
4545
include(${PROJECT_ROOT_DIR}/cmake/bazel.cmake)
4646
include(${PROJECT_ROOT_DIR}/cmake/option.cmake)
4747

48-
if(NOT ANDROID AND AUTO_DETECT_ARCH AND CMAKE_SYSTEM_PROCESSOR MATCHES "x86_64|i686|i386|x64")
48+
if (NOT ANDROID AND AUTO_DETECT_ARCH AND HOST_ARCH MATCHES "^(x86|x64)$")
4949
setup_compiler_march_for_x86(MATH_MARCH_FLAG_SSE MATH_MARCH_FLAG_AVX2 MATH_MARCH_FLAG_AVX512 MATH_MARCH_FLAG_AVX512FP16)
5050
message(STATUS "best compiler march, sse: " ${MATH_MARCH_FLAG_SSE} ", avx2: " ${MATH_MARCH_FLAG_AVX2} ", avx512: " ${MATH_MARCH_FLAG_AVX512} ", avx512fp16: " ${MATH_MARCH_FLAG_AVX512FP16})
5151
endif()

src/ailego/CMakeLists.txt

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -19,12 +19,9 @@ if(UNIX AND NOT APPLE)
1919
endif()
2020

2121
if(NOT ANDROID AND AUTO_DETECT_ARCH)
22-
message(WARNING "before setup_compiler_march_for_x86")
23-
message(WARNING "before CMAKE_SYSTEM_PROCESSOR - ${CMAKE_SYSTEM_PROCESSOR}")
24-
message(WARNING "before HOST_ARCH - ${HOST_ARCH}")
2522
if(HOST_ARCH MATCHES "^(x86|x64)$")
2623
setup_compiler_march_for_x86(MATH_MARCH_FLAG_SSE MATH_MARCH_FLAG_AVX2 MATH_MARCH_FLAG_AVX512 MATH_MARCH_FLAG_AVX512FP16)
27-
message(STATUS "best compiler march, sse: " ${MATH_MARCH_FLAG_SSE} ", avx2: " ${MATH_MARCH_FLAG_AVX2} ", avx512: " ${MATH_MARCH_FLAG_AVX512} ", avx512fp16: " ${MATH_MARCH_FLAG_AVX512FP16})
24+
message(STATUS "best compiler march - sse: " ${MATH_MARCH_FLAG_SSE} ", avx2: " ${MATH_MARCH_FLAG_AVX2} ", avx512: " ${MATH_MARCH_FLAG_AVX512} ", avx512fp16: " ${MATH_MARCH_FLAG_AVX512FP16})
2825

2926
file(GLOB_RECURSE MATH_FILES_SSE
3027
${CMAKE_CURRENT_SOURCE_DIR}/math/*_sse.cc
@@ -93,7 +90,7 @@ if(NOT ANDROID AND AUTO_DETECT_ARCH)
9390
COMPILE_FLAGS "${MATH_MARCH_FLAG_AVX512FP16}"
9491
)
9592
endforeach()
96-
elseif(HOST_ARCH MATCHES "^(x86|x64)$")
93+
elseif (HOST_ARCH MATCHES "^(arm|arm64)$")
9794
if(MSVC)
9895
return()
9996
endif()

src/turbo/CMakeLists.txt

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,9 +2,9 @@ include(${PROJECT_ROOT_DIR}/cmake/bazel.cmake)
22
include(${PROJECT_ROOT_DIR}/cmake/option.cmake)
33

44
if(NOT ANDROID AND AUTO_DETECT_ARCH)
5-
if(CMAKE_SYSTEM_PROCESSOR MATCHES "x86_64|i686|i386|x64")
5+
if (HOST_ARCH MATCHES "^(x86|x64)$")
66
setup_compiler_march_for_x86(TURBO_MARCH_FLAG_SSE TURBO_MARCH_FLAG_AVX2 TURBO_MARCH_FLAG_AVX512 TURBO_MARCH_FLAG_AVX512FP16)
7-
elseif(CMAKE_SYSTEM_PROCESSOR MATCHES "aarch64|arm64|ARM64")
7+
elseif (HOST_ARCH MATCHES "^(arm|arm64)$")
88
# ARM64 architecture - no special march flags needed for now
99
# NEON implementations can be added here if needed
1010
message(STATUS "turbo: ARM64 detected, skipping x86-specific optimizations")
@@ -18,7 +18,7 @@ file(GLOB_RECURSE ALL_SRCS *.cc *.c *.h)
1818
# same directory that adds the sources to a target (i.e. here, not in a
1919
# subdirectory).
2020
if(NOT ANDROID AND AUTO_DETECT_ARCH)
21-
if(CMAKE_SYSTEM_PROCESSOR MATCHES "x86_64|i686|i386|x64")
21+
if (HOST_ARCH MATCHES "^(x86|x64)$")
2222
file(GLOB_RECURSE AVX512_VNNI_SRCS ${CMAKE_CURRENT_SOURCE_DIR}/avx512_vnni/*.cc)
2323
set_source_files_properties(
2424
${AVX512_VNNI_SRCS}

0 commit comments

Comments
 (0)