commit c29f51642ecc41da3a6a40e7685ef4decce8bbf3
author Nikita Shulga <nikita.shulga@gmail.com> Tue Dec 08 07:52:52 2020 -0800
committer Facebook GitHub Bot <facebook-github-bot@users.noreply.github.com> Tue Dec 08 07:58:22 2020 -0800
tree eac006a1e2a3a052c508ffb370be75df2609f6c0
parent 58c13cf685969bec4d49848399cd72c8f6834857

Modify NEON check for ARM64 on OS X (#48982)

Summary:
Use CMAKE_SYSTEM_PROCESSOR rather than run sysctl

Fixes https://github.com/pytorch/pytorch/issues/48874

Pull Request resolved: https://github.com/pytorch/pytorch/pull/48982

Reviewed By: walterddr

Differential Revision: D25385883

Pulled By: malfet

fbshipit-source-id: 47b6dc5be8d75f6d4a66a11c564abdfe31ac90b4

cmake/Modules/FindARM.cmake

diff --git a/cmake/Modules/FindARM.cmake b/cmake/Modules/FindARM.cmake
index bd68f5f..acd00cf 100644
--- a/cmake/Modules/FindARM.cmake
+++ b/cmake/Modules/FindARM.cmake
@@ -41,9 +41,7 @@
    ENDIF (OMAP4_TRUE)
 
 ELSEIF(CMAKE_SYSTEM_NAME MATCHES "Darwin")
-   EXEC_PROGRAM("/usr/sbin/sysctl -n hw.optional.arm64" OUTPUT_VARIABLE
-      IS_ARM64)
-   IF(IS_ARM64 STREQUAL "1")
+   IF(CMAKE_SYSTEM_PROCESSOR STREQUAL "arm64")
       set(NEON_FOUND true CACHE BOOL "NEON available on ARM64")
    ENDIF()
    EXEC_PROGRAM("/usr/sbin/sysctl -n machdep.cpu.features" OUTPUT_VARIABLE