Skip to content

Commit

Permalink
revert many-rocm wheel build when getting CI signal; add comment abou…
Browse files Browse the repository at this point in the history
…t runner
  • Loading branch information
tenpercent committed Sep 4, 2024
1 parent 146a97c commit 3721f76
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions .github/workflows/wheels.yml
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,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_v) for rocm_v in ROCM_VERSIONS]
PY_ROCM = [(PY_VERSIONS[-1], ROCM_VERSIONS[-1])]
include = []
for os in ['8-core-ubuntu', 'windows-8-core']:
Expand All @@ -60,7 +60,7 @@ jobs:
if os == 'windows-8-core':
continue
include.append(dict(
os="16-core-ubuntu",
os="16-core-ubuntu", # use for ROCm wheels only to avoid CI timeouts
python=python,
torch_version=torch_version,
toolkit_type="rocm",
Expand Down

0 comments on commit 3721f76

Please sign in to comment.