ci: rocm builds (#9796)

This commit is contained in:
harakas 2024-02-12 14:24:02 +02:00 committed by GitHub
parent b9e20fc833
commit 09153a154a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -47,6 +47,57 @@ jobs:
tensorrt.tags=${{ steps.setup.outputs.image-name }}-tensorrt tensorrt.tags=${{ steps.setup.outputs.image-name }}-tensorrt
*.cache-from=type=registry,ref=${{ steps.setup.outputs.cache-name }}-amd64 *.cache-from=type=registry,ref=${{ steps.setup.outputs.cache-name }}-amd64
*.cache-to=type=registry,ref=${{ steps.setup.outputs.cache-name }}-amd64,mode=max *.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: arm64_build:
runs-on: ubuntu-latest runs-on: ubuntu-latest
name: ARM Build name: ARM Build