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
2026-07-20 18:59:01 +03:00 · 2025-03-02 21:46:46 -06:00
parent b8f4cb5435
commit 0128ec2ba6
9 changed files with 36 additions and 670 deletions
@@ -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