Skip to content

Commit c667185

Browse files
committed
DEBUG: only CUDA 13.0 with py3.13 and one GPU arch
1 parent fbca247 commit c667185

File tree

3 files changed

+8
-2
lines changed

3 files changed

+8
-2
lines changed

recipe/bld.bat

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -98,7 +98,7 @@ if not "%cuda_compiler_version%" == "None" (
9898
if "%cuda_compiler_version:~0,2%"=="12" (
9999
set "TORCH_CUDA_ARCH_LIST=5.0;6.0;7.0;7.5;8.0;8.6;9.0;10.0;12.0+PTX"
100100
) else if "%cuda_compiler_version%" == "13.0" (
101-
set "TORCH_CUDA_ARCH_LIST=7.5;8.0;8.6;9.0;10.0;11.0;12.0+PTX"
101+
set "TORCH_CUDA_ARCH_LIST=7.5+PTX"
102102
REM c.f. https://github.com/pytorch/pytorch/pull/161316
103103
set "TORCH_NVCC_FLAGS=!TORCH_NVCC_FLAGS! -compress-mode=size"
104104
) else (

recipe/build.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -230,7 +230,7 @@ elif [[ ${cuda_compiler_version} != "None" ]]; then
230230
export TORCH_CUDA_ARCH_LIST="5.0;6.0;7.0;7.5;8.0;8.6;9.0;10.0;12.0+PTX"
231231
;;
232232
13.0)
233-
export TORCH_CUDA_ARCH_LIST="7.5;8.0;8.6;9.0;10.0;11.0;12.0+PTX"
233+
export TORCH_CUDA_ARCH_LIST="7.5+PTX"
234234
# c.f. https://github.com/pytorch/pytorch/pull/161316
235235
export TORCH_NVCC_FLAGS="$TORCH_NVCC_FLAGS -compress-mode=size"
236236
;;

recipe/meta.yaml

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -67,6 +67,8 @@ source:
6767

6868
build:
6969
number: {{ build }}
70+
# debug
71+
skip: true # [cuda_compiler_version != "13.0"]
7072
# This logic allows two rc variants to be defined in the conda_build_config, but only one to actually be built.
7173
# We want to be able to define two variants in the cbc so we can assign different labels to each in the upload channel
7274
# (by zipping is_rc with channel_targets). This prevents rc builds being used unless specifically requested.
@@ -237,6 +239,8 @@ outputs:
237239
string: cuda{{ cuda_compiler_version | replace('.', '') }}_{{ blas_impl }}_py{{ CONDA_PY }}_h{{ PKG_HASH }}_{{ build }} # [cuda_compiler_version != "None"]
238240
string: cpu_{{ blas_impl }}_py{{ CONDA_PY }}_h{{ PKG_HASH }}_{{ build }} # [cuda_compiler_version == "None"]
239241
detect_binary_files_with_prefix: false
242+
# debug
243+
skip: true # [py!=312]
240244
run_exports:
241245
- {{ pin_subpackage('pytorch', max_pin='x.x') }}
242246
- {{ pin_subpackage('libtorch', max_pin='x.x') }}
@@ -341,6 +345,8 @@ outputs:
341345
string: cuda{{ cuda_compiler_version | replace('.', '') }}_{{ blas_impl }}_py{{ CONDA_PY }}_h{{ PKG_HASH }}_{{ build }} # [cuda_compiler_version != "None"]
342346
string: cpu_{{ blas_impl }}_py{{ CONDA_PY }}_h{{ PKG_HASH }}_{{ build }} # [cuda_compiler_version == "None"]
343347
detect_binary_files_with_prefix: false
348+
# debug
349+
skip: true # [py!=312]
344350
requirements:
345351
host:
346352
- python

0 commit comments

Comments
 (0)