Upgrade RocM to 6.3.3 (#16900)

* Simplify rocm install and update to 6.3.1

* Build out more necessary packages

* Update to 6.3.3

* Set bake version

* Fix typo

* Ensure NHWC is used

* Reset dev changes

* Write to cache
This commit is contained in:
Nicolas Mowen
2025-03-02 20:46:46 -07:00
committed by GitHub
parent b8f4cb5435
commit 0128ec2ba6
9 changed files with 36 additions and 670 deletions

View File

@@ -175,6 +175,7 @@ jobs:
files: docker/rocm/rocm.hcl
set: |
rocm.tags=${{ steps.setup.outputs.image-name }}-rocm
*.cache-to=type=registry,ref=${{ steps.setup.outputs.cache-name }}-rocm,mode=max
*.cache-from=type=gha
arm64_extra_builds:
runs-on: ubuntu-22.04-arm