Skip to content

Commit

Permalink
Turn off custom and optimized kernels for mac x64
Browse files Browse the repository at this point in the history
  • Loading branch information
zcbenz committed Nov 14, 2024
1 parent f180d69 commit d13eb8f
Showing 1 changed file with 2 additions and 5 deletions.
7 changes: 2 additions & 5 deletions CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -34,10 +34,8 @@ if(TORCH_BACKEND_ALL)
endif()

# https://github.com/pytorch/executorch/issues/6839
if(CMAKE_SYSTEM_NAME MATCHES "Darwin")
if(CMAKE_SYSTEM_PROCESSOR MATCHES "x86_64")
set(TORCH_KERNELS_CUSTOM=OFF)
endif()
if(CMAKE_SYSTEM_NAME MATCHES "Darwin" AND CMAKE_SYSTEM_PROCESSOR MATCHES "x86_64")
set(TORCH_KERNELS_CUSTOM OFF)
endif()

# ----------------------------- Node-API -------------------------------
Expand Down Expand Up @@ -181,6 +179,5 @@ message(STATUS " TORCH_BACKEND_MPS : ${TORCH_BACKEND_MPS}")
message(STATUS " TORCH_BACKEND_XNNPACK : ${TORCH_BACKEND_XNNPACK}")
message(STATUS " TORCH_KERNELS_CUSTOM : ${TORCH_KERNELS_CUSTOM}")
message(STATUS " TORCH_KERNELS_OPTIMIZED : ${TORCH_KERNELS_OPTIMIZED}")
message(STATUS " TORCH_KERNELS_PORTABLE : ${TORCH_KERNELS_PORTABLE}")
message(STATUS " TORCH_KERNELS_QUANTIZED : ${TORCH_KERNELS_QUANTIZED}")
message(STATUS "")

0 comments on commit d13eb8f

Please sign in to comment.