Try again

conda-forge · Feb 1, 2025 · 3811a1c · 3811a1c
1 parent 6fbe6c8
commit 3811a1c
Showing 1 changed file with 34 additions and 7 deletions.
diff --git a/recipe/patch_yaml/pytorch.yaml b/recipe/patch_yaml/pytorch.yaml
@@ -174,15 +174,42 @@ if:
   name: pytorch-gpu
   version: 2.4.1
   subdir_in: linux-64
-  build_number_in: [0, 1, 2, 3, 4, 5, 100, 101, 102, 103, 104, 105, 200, 201, 202, 203, 204, 205, 300, 301, 302, 303, 304, 305]
+  build_number_in:
+    - 0
+    - 1
+    - 2
+    - 3
+    - 4
+    - 5
+    # mkl is +100
+    - 100
+    - 101
+    - 102
+    - 103
+    - 104
+    - 105
+    # cuda compiler is +200
+    - 200
+    - 201
+    - 202
+    - 203
+    - 204
+    - 205
+    # cuda compiler + mkl is +300
+    - 300
+    - 301
+    - 302
+    - 303
+    - 304
+    - 305
 
 then:
   - replace_depends:
-    old: pytorch ${version} cuda118_*${build_number}
-    new: pytorch ${version} cuda118_*${build_number}
+    old: pytorch 2.4.1 cuda118_*
+    new: pytorch 2.4.1 cuda118_*${build_number}
   - replace_depends:
-    old: pytorch ${version} cuda120_*${build_number}
-    new: pytorch ${version} cuda120_*${build_number}
+    old: pytorch 2.4.1 cuda120_*
+    new: pytorch 2.4.1 cuda120_*${build_number}
   - replace_depends:
-    old: pytorch ${version} cpu_
-    new: pytorch ${version} cpu_*${build_number}
+    old: pytorch 2.4.1 cpu_*
+    new: pytorch 2.4.1 cpu_*${build_number}