Remove benchmark dependencies from regular ROCm CI images#160047
Remove benchmark dependencies from regular ROCm CI images#160047huydhn wants to merge 2 commits intopytorch:mainfrom
Conversation
Signed-off-by: Huy Do <huydhn@gmail.com>
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/160047
Note: Links to docs will display an error until the docs builds have been completed. ⏳ No Failures, 1 PendingAs of commit a2639c9 with merge base 2507ae6 ( This comment was automatically generated by Dr. CI and updates every 15 minutes. |
| @@ -192,6 +192,21 @@ case "$tag" in | |||
| UCC_COMMIT=${_UCC_COMMIT} | |||
| INDUCTOR_BENCHMARKS=yes | |||
There was a problem hiding this comment.
Works, although could be made more concise by having a condition like: if [[ $tag =~ "benchmarks" ]]; to avoid repetition
Signed-off-by: Huy Do <huydhn@gmail.com>
|
@pytorchbot merge -f 'ROCm signals looks ok' |
Merge startedYour change will be merged immediately since you used the force (-f) flag, bypassing any CI checks (ETA: 1-5 minutes). Please use Learn more about merging in the wiki. Questions? Feedback? Please reach out to the PyTorch DevX Team |
Follow-up to pytorch#160047 that separated the rocm image into default CI and benchmarks.
Follow-up to #160047 that separated the rocm image into default CI and benchmarks. Pull Request resolved: #160421 Approved by: https://github.com/jeffdaily Co-authored-by: Jeff Daily <jeff.daily@amd.com>
Follow-up to #160047 that separated the rocm image into default CI and benchmarks. Pull Request resolved: #160421 Approved by: https://github.com/jeffdaily Co-authored-by: Jeff Daily <jeff.daily@amd.com>
Follow-up to #160047 that separated the rocm image into default CI and benchmarks. Pull Request resolved: #160421 Approved by: https://github.com/jeffdaily Co-authored-by: Jeff Daily <jeff.daily@amd.com>
Follow-up to pytorch#160047 that separated the rocm image into default CI and benchmarks. Pull Request resolved: pytorch#160421 Approved by: https://github.com/jeffdaily Co-authored-by: Jeff Daily <jeff.daily@amd.com>
…0047) Instead, use a new `pytorch-linux-jammy-rocm-n-py3-benchmarks` image for Docker benchmark job. This addresses 2 issues: * The current ROCm failures in trunk w.r.t librosa version https://github.com/pytorch/pytorch/actions/runs/16789466749/job/47549950994 that TorchBench pulls in. * Reduce the size of the regular ROCm CI images by removing TorchBench models, which is needed only for benchmarking jobs. Pull Request resolved: pytorch#160047 Approved by: https://github.com/malfet, https://github.com/izaitsevfb
Follow-up to pytorch#160047 that separated the rocm image into default CI and benchmarks. Pull Request resolved: pytorch#160421 Approved by: https://github.com/jeffdaily Co-authored-by: Jeff Daily <jeff.daily@amd.com>
Instead, use a new
pytorch-linux-jammy-rocm-n-py3-benchmarksimage for Docker benchmark job. This addresses 2 issues:cc @jeffdaily @sunway513 @jithunnair-amd @pruthvistony @ROCmSupport @dllehr-amd @jataylo @hongxiayang @naromero77amd