We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
onnxruntime_NVCC_THREADS
1 parent 798cea2 commit 5bfca1dCopy full SHA for 5bfca1d
cmake/onnxruntime_providers_cuda.cmake
@@ -153,7 +153,7 @@
153
# CUDA 11.3+ supports parallel compilation
154
# https://docs.nvidia.com/cuda/cuda-compiler-driver-nvcc/index.html#options-for-guiding-compiler-driver-threads
155
if (CMAKE_CUDA_COMPILER_VERSION VERSION_GREATER_EQUAL 11.3)
156
- option(onnxruntime_NVCC_THREADS "Number of threads that NVCC can use for compilation." 1)
+ set(onnxruntime_NVCC_THREADS "1" CACHE STRING "Number of threads that NVCC can use for compilation.")
157
target_compile_options(${target} PRIVATE "$<$<COMPILE_LANGUAGE:CUDA>:SHELL:--threads \"${onnxruntime_NVCC_THREADS}\">")
158
endif()
159
0 commit comments