ci: rocm builds (#9796)

blakeblackshear · Feb 12, 2024 · 09153a1 · 09153a1
1 parent b9e20fc
commit 09153a1
Showing 1 changed file with 51 additions and 0 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -47,6 +47,57 @@ jobs:
  tensorrt.tags=${{ steps.setup.outputs.image-name }}-tensorrt
  *.cache-from=type=registry,ref=${{ steps.setup.outputs.cache-name }}-amd64
  *.cache-to=type=registry,ref=${{ steps.setup.outputs.cache-name }}-amd64,mode=max
+ - name: AMD/ROCm general build
+ env:
+ AMDGPU: gfx
+ HSA_OVERRIDE: 0
+ uses: docker/bake-action@v3
+ with:
+ push: true
+ targets: rocm
+ files: docker/rocm/rocm.hcl
+ set: |
+ rocm.tags=${{ steps.setup.outputs.image-name }}-rocm
+ *.cache-from=type=gha
+ - name: AMD/ROCm gfx900
+ env:
+ AMDGPU: gfx900
+ HSA_OVERRIDE: 1
+ HSA_OVERRIDE_GFX_VERSION: 9.0.0
+ uses: docker/bake-action@v3
+ with:
+ push: true
+ targets: rocm
+ files: docker/rocm/rocm.hcl
+ set: |
+ rocm.tags=${{ steps.setup.outputs.image-name }}-rocm-gfx900
+ *.cache-from=type=gha
+ - name: AMD/ROCm gfx1030
+ env:
+ AMDGPU: gfx1030
+ HSA_OVERRIDE: 1
+ HSA_OVERRIDE_GFX_VERSION: 10.3.0
+ uses: docker/bake-action@v3
+ with:
+ push: true
+ targets: rocm
+ files: docker/rocm/rocm.hcl
+ set: |
+ rocm.tags=${{ steps.setup.outputs.image-name }}-rocm-gfx1030
+ *.cache-from=type=gha
+ - name: AMD/ROCm gfx1100
+ env:
+ AMDGPU: gfx1100
+ HSA_OVERRIDE: 1
+ HSA_OVERRIDE_GFX_VERSION: 11.0.0
+ uses: docker/bake-action@v3
+ with:
+ push: true
+ targets: rocm
+ files: docker/rocm/rocm.hcl
+ set: |
+ rocm.tags=${{ steps.setup.outputs.image-name }}-rocm-gfx1100
+ *.cache-from=type=gha
  arm64_build:
  runs-on: ubuntu-latest
  name: ARM Build