Skip to content

feat(rocm): upgrade GPU base images to ROCm 10.0 - #194

Open
kerwtsai-amd wants to merge 2 commits into
developfrom
feat/rocm-10-base-image
Open

feat(rocm): upgrade GPU base images to ROCm 10.0#194
kerwtsai-amd wants to merge 2 commits into
developfrom
feat/rocm-10-base-image

Conversation

@kerwtsai-amd

Copy link
Copy Markdown
Collaborator

Summary

  • Upgrade the GPU base image from ROCm 7.13 to ROCm 10.0 (amdrocm-core-sdk10.0-* + torch[device-*]==2.13.0+rocm10.0.0 from stable.repo.amd.com).
  • Keep the existing public GPU_TARGET image tags. Family buckets expand internally: gfx110x → gfx1100/1101/1102/1103, gfx120x → gfx1200/gfx1201.
  • Add a prebuilt gfx1153 target to the Docker build matrix so CI publishes auplc-base:*-gfx1153.

Test plan

  • Confirm this PR triggers Build Docker Images against develop (base-gpu matrix should include gfx110x, gfx1150, gfx1151, gfx1152, gfx1153, gfx120x).
  • Confirm each base-gpu job installs the matching amdrocm-core-sdk10.0-<isa> packages and PyTorch extras from https://stable.repo.amd.com/rocm/whl-next/.
  • Confirm Python unit tests (tests/installer) pass, including the new ROCm 10 / gfx1153 mapping checks.
  • After images publish, spot-check auplc-base:*-gfx1151 (Strix Halo / MAX Pro 395) and auplc-base:*-gfx1153.

Made with Cursor

Keep the existing GPU_TARGET image tags, expand gfx110x/gfx120x to the concrete ROCm 10 ISAs, and add a prebuilt gfx1153 target so Ryzen AI 300/400 iGPUs can pull a matching wheel set.
The ROCm 10 ISA lookup lines exceeded the 200-character yamllint limit and failed YAML (yamllint) on the PR.
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.

1 participant