|
66 | 66 | {% else %} |
67 | 67 | skip: true # [is_rc] |
68 | 68 | {% endif %} |
69 | | - string: cuda{{ cuda_compiler_version | replace('.', '') }}_{{ blas_impl }}_h{{ PKG_HASH }}_{{ PKG_BUILDNUM }} # [cuda_compiler_version != "None"] |
70 | | - string: cpu_{{ blas_impl }}_h{{ PKG_HASH }}_{{ PKG_BUILDNUM }} # [cuda_compiler_version == "None"] |
| 69 | + string: cuda{{ cuda_compiler_version | replace('.', '') }}_{{ blas_impl }}_h{{ PKG_HASH }}_{{ build }} # [cuda_compiler_version != "None"] |
| 70 | + string: cpu_{{ blas_impl }}_h{{ PKG_HASH }}_{{ build }} # [cuda_compiler_version == "None"] |
71 | 71 | detect_binary_files_with_prefix: false |
72 | 72 | run_exports: |
73 | 73 | - {{ pin_subpackage('libtorch', max_pin='x.x') }} |
@@ -168,8 +168,8 @@ requirements: |
168 | 168 | - pytorch-gpu ==99999999 # [cuda_compiler_version == "None"] |
169 | 169 | - pytorch-gpu =={{ version }} # [cuda_compiler_version != "None"] |
170 | 170 | - pytorch-cpu ==99999999 # [cuda_compiler_version != "None"] |
171 | | - - pytorch {{ version }} cuda{{ cuda_compiler_version | replace('.', '') }}_{{ blas_impl }}_*_{{ PKG_BUILDNUM }} # [cuda_compiler_version != "None"] |
172 | | - - pytorch {{ version }} cpu_{{ blas_impl }}_*_{{ PKG_BUILDNUM }} # [cuda_compiler_version == "None"] |
| 171 | + - pytorch {{ version }} cuda{{ cuda_compiler_version | replace('.', '') }}_{{ blas_impl }}_*_{{ build }} # [cuda_compiler_version != "None"] |
| 172 | + - pytorch {{ version }} cpu_{{ blas_impl }}_*_{{ build }} # [cuda_compiler_version == "None"] |
173 | 173 | # if using OpenBLAS, ensure that a version compatible with OpenMP is used |
174 | 174 | # otherwise, we get the following warnings: |
175 | 175 | # OpenBLAS Warning : Detect OpenMP Loop and this application may hang. Please rebuild the library with USE_OPENMP=1 option. |
@@ -216,8 +216,8 @@ outputs: |
216 | 216 | script: build.sh # [unix] |
217 | 217 | script: bld.bat # [win] |
218 | 218 | build: |
219 | | - string: cuda{{ cuda_compiler_version | replace('.', '') }}_{{ blas_impl }}_py{{ CONDA_PY }}_h{{ PKG_HASH }}_{{ PKG_BUILDNUM }} # [cuda_compiler_version != "None"] |
220 | | - string: cpu_{{ blas_impl }}_py{{ CONDA_PY }}_h{{ PKG_HASH }}_{{ PKG_BUILDNUM }} # [cuda_compiler_version == "None"] |
| 219 | + string: cuda{{ cuda_compiler_version | replace('.', '') }}_{{ blas_impl }}_py{{ CONDA_PY }}_h{{ PKG_HASH }}_{{ build }} # [cuda_compiler_version != "None"] |
| 220 | + string: cpu_{{ blas_impl }}_py{{ CONDA_PY }}_h{{ PKG_HASH }}_{{ build }} # [cuda_compiler_version == "None"] |
221 | 221 | detect_binary_files_with_prefix: false |
222 | 222 | run_exports: |
223 | 223 | - {{ pin_subpackage('pytorch', max_pin='x.x') }} |
@@ -440,19 +440,19 @@ outputs: |
440 | 440 | {% set pytorch_cpu_gpu = "pytorch-gpu" %} # [cuda_compiler_version != "None"] |
441 | 441 | - name: {{ pytorch_cpu_gpu }} |
442 | 442 | build: |
443 | | - string: cuda{{ cuda_compiler_version | replace('.', '') }}_{{ blas_impl }}_h{{ PKG_HASH }}_{{ PKG_BUILDNUM }} # [megabuild and cuda_compiler_version != "None"] |
444 | | - string: cpu_{{ blas_impl }}_h{{ PKG_HASH }}_{{ PKG_BUILDNUM }} # [megabuild and cuda_compiler_version == "None"] |
445 | | - string: cuda{{ cuda_compiler_version | replace('.', '') }}_{{ blas_impl }}py{{ CONDA_PY }}_h{{ PKG_HASH }}_{{ PKG_BUILDNUM }} # [not megabuild and cuda_compiler_version != "None"] |
446 | | - string: cpu_{{ blas_impl }}_py{{ CONDA_PY }}_h{{ PKG_HASH }}_{{ PKG_BUILDNUM }} # [not megabuild and cuda_compiler_version == "None"] |
| 443 | + string: cuda{{ cuda_compiler_version | replace('.', '') }}_{{ blas_impl }}_h{{ PKG_HASH }}_{{ build }} # [megabuild and cuda_compiler_version != "None"] |
| 444 | + string: cpu_{{ blas_impl }}_h{{ PKG_HASH }}_{{ build }} # [megabuild and cuda_compiler_version == "None"] |
| 445 | + string: cuda{{ cuda_compiler_version | replace('.', '') }}_{{ blas_impl }}py{{ CONDA_PY }}_h{{ PKG_HASH }}_{{ build }} # [not megabuild and cuda_compiler_version != "None"] |
| 446 | + string: cpu_{{ blas_impl }}_py{{ CONDA_PY }}_h{{ PKG_HASH }}_{{ build }} # [not megabuild and cuda_compiler_version == "None"] |
447 | 447 | detect_binary_files_with_prefix: false |
448 | 448 | # weigh down cpu implementation and give cuda preference |
449 | 449 | track_features: |
450 | 450 | - pytorch-cpu # [cuda_compiler_version == "None"] |
451 | 451 | requirements: |
452 | 452 | run: |
453 | | - - pytorch {{ version }}=cuda*_{{ blas_impl }}*{{ PKG_BUILDNUM }} # [megabuild and cuda_compiler_version != "None"] |
454 | | - - pytorch {{ version }}=cpu_{{ blas_impl }}*{{ PKG_BUILDNUM }} # [megabuild and cuda_compiler_version == "None"] |
455 | | - - {{ pin_subpackage("pytorch", exact=True) }} # [not megabuild] |
| 453 | + - pytorch {{ version }}=cuda*_{{ blas_impl }}*{{ build }} # [megabuild and cuda_compiler_version != "None"] |
| 454 | + - pytorch {{ version }}=cpu_{{ blas_impl }}*{{ build }} # [megabuild and cuda_compiler_version == "None"] |
| 455 | + - {{ pin_subpackage("pytorch", exact=True) }} # [not megabuild] |
456 | 456 | test: |
457 | 457 | imports: |
458 | 458 | - torch |
|
0 commit comments