Update wheels.yml

facebookresearch · Jan 16, 2025 · b1c7e73 · b1c7e73
1 parent 9f6c372
commit b1c7e73
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/wheels.yml b/.github/workflows/wheels.yml
@@ -42,7 +42,7 @@ jobs:
           for cu in CU_VERSIONS[1:]:
             PY_CU.append((PY_VERSIONS[-1], cu))
           print("Limited matrix PY_CU", PY_CU)
-          PY_ROCM = [(PY_VERSIONS[-1], ROCM_VERSIONS[-1])]
+          PY_ROCM = [(PY_VERSIONS[-1], ROCM_VERSIONS[-2])] # Always last version is upcoming version
 
         include = []
         for os in ['8-core-ubuntu', 'windows-8-core']: