Allow CUDA override only during package build

jeongseok-meta · Dec 6, 2024 · 4013c11 · 4013c11
1 parent 43fc636
commit 4013c11
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/recipe/build.sh b/recipe/build.sh
@@ -2,6 +2,8 @@
 
 set -ex
 
+export IN_PYTORCH_BUILD
+
 # https://github.com/conda-forge/pytorch-cpu-feedstock/issues/243
 # https://github.com/pytorch/pytorch/blob/v2.3.1/setup.py#L341
 export PACKAGE_TYPE=conda

diff --git a/recipe/patches/0010-Allow-overriding-CUDA-related-paths.patch b/recipe/patches/0010-Allow-overriding-CUDA-related-paths.patch
@@ -17,7 +17,7 @@ index ec9ae530..76d1fb9a 100644
  # We compute the rest based on those here to avoid re-searching and to avoid finding a possibly
  # different installation.
 -if(CMAKE_CUDA_COMPILER_TOOLKIT_ROOT)
-+if(FALSE)
++if(CMAKE_CUDA_COMPILER_TOOLKIT_ROOT AND NOT $ENV{IN_PYTORCH_BUILD})
    set(CUDAToolkit_ROOT_DIR "${CMAKE_CUDA_COMPILER_TOOLKIT_ROOT}")
    set(CUDAToolkit_LIBRARY_ROOT "${CMAKE_CUDA_COMPILER_LIBRARY_ROOT}")
    set(CUDAToolkit_VERSION "${CMAKE_CUDA_COMPILER_TOOLKIT_VERSION}")