Skip to content

Commit a4a1f24

Browse files
committed
no custom or quantized ops on win cuda
1 parent 98baab7 commit a4a1f24

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

tools/cmake/preset/llm.cmake

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,9 +26,13 @@ if(CMAKE_SYSTEM_NAME STREQUAL "Darwin")
2626
elseif(CMAKE_SYSTEM_NAME STREQUAL "Linux")
2727
# Linux-specific code here
2828
elseif(CMAKE_SYSTEM_NAME STREQUAL "Windows" OR CMAKE_SYSTEM_NAME STREQUAL
29-
"WIN32"
29+
"WIN32"
3030
)
31-
# Windows or other OS-specific code here
31+
# Windows-specific code: disable quantized and custom ops when building with CUDA
32+
if(EXECUTORCH_BUILD_CUDA)
33+
set_overridable_option(EXECUTORCH_BUILD_KERNELS_QUANTIZED OFF)
34+
set_overridable_option(EXECUTORCH_BUILD_KERNELS_LLM OFF)
35+
endif()
3236
elseif(CMAKE_SYSTEM_NAME STREQUAL "Android")
3337
# Android-specific code here
3438
else()

0 commit comments

Comments
 (0)