Skip to content

remove the static pytorch-rocm-requirements.txt, always use generator#1147

Merged
willghatch merged 1 commit intomainfrom
users/willghatch/remove-static-rocm-require
Mar 24, 2026
Merged

remove the static pytorch-rocm-requirements.txt, always use generator#1147
willghatch merged 1 commit intomainfrom
users/willghatch/remove-static-rocm-require

Conversation

@willghatch
Copy link
Copy Markdown
Contributor

The static file hardcoded ROCm 6.4, which breaks on other ROCm versions. Use gen-pytorch-rocm-requirements.py everywhere (IE in documentation, CI) instead, which auto-detects the installed ROCm version and probes for a matching PyTorch wheel source.

This is a follow-up to the discussion in #1134

@willghatch willghatch requested review from ftynse and tgymnich March 18, 2026 23:51
@willghatch willghatch force-pushed the users/willghatch/remove-static-rocm-require branch from a55d202 to 991f94a Compare March 19, 2026 17:12
The static file hardcoded ROCm 6.4, which breaks on other ROCm versions.
Use gen-pytorch-rocm-requirements.py everywhere (IE in documentation, CI) instead, which auto-detects the installed ROCm version and probes for a matching PyTorch wheel source.
Also adjust generator version regex to allow build number.

This is a follow-up to the discussion in #1134

Signed-off-by: William G Hatch <william@hatch.uno>
@willghatch willghatch force-pushed the users/willghatch/remove-static-rocm-require branch from 991f94a to 7d04f83 Compare March 19, 2026 20:57
@willghatch
Copy link
Copy Markdown
Contributor Author

@ftynse @tgymnich I don't know where our flakiness comes from, but this and #1155 are both passing all-green, after re-running flakey tests that failed.

Copy link
Copy Markdown
Contributor

@ftynse ftynse left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

🤷 maybe torch version changed in the meantime

@willghatch willghatch merged commit f223c86 into main Mar 24, 2026
19 checks passed
@willghatch willghatch deleted the users/willghatch/remove-static-rocm-require branch March 24, 2026 17:23
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants