Enable manywheel build and smoke test on main branch for ROCm (#153287)

Fixes issue of not discovering breakage of ROCm wheel builds until the nightly job runs e.g. https://github.com/pytorch/pytorch/pull/153253

Pull Request resolved: https://github.com/pytorch/pytorch/pull/153287
Approved by: https://github.com/jeffdaily
This commit is contained in:
Jithun Nair
2025-06-14 19:14:31 +00:00
committed by PyTorch MergeBot
parent 5285d10243
commit 794ef6c9b8
6 changed files with 109 additions and 16 deletions

View File

@ -152,7 +152,7 @@ LINUX_BINARY_SMOKE_WORKFLOWS = [
package_type="manywheel",
build_configs=generate_binary_build_matrix.generate_wheels_matrix(
OperatingSystem.LINUX,
arches=["12.6", "12.8", "12.9"],
arches=["12.6", "12.8", "12.9", "6.4"],
python_versions=["3.9"],
),
branches="main",