Do not assume NEON is present when cross-compiling for x86_64

pytorch · Dec 21, 2020 · 9a1af6f · 9a1af6f
1 parent aea336d
commit 9a1af6f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cmake/Modules/FindARM.cmake b/cmake/Modules/FindARM.cmake
@@ -41,7 +41,7 @@ IF(CMAKE_SYSTEM_NAME MATCHES "Linux")
    ENDIF (OMAP4_TRUE)
 
 ELSEIF(CMAKE_SYSTEM_NAME MATCHES "Darwin")
-   IF(CMAKE_SYSTEM_PROCESSOR STREQUAL "arm64")
+   IF(CMAKE_SYSTEM_PROCESSOR STREQUAL "arm64" AND NOT CMAKE_OSX_ARCHITECTURES STREQUAL "x86_64")
       set(NEON_FOUND true CACHE BOOL "NEON available on ARM64")
    ENDIF()
    EXEC_PROGRAM("/usr/sbin/sysctl -n machdep.cpu.features" OUTPUT_VARIABLE