Skip to content

Commit e1c8db3

Browse files
steevecerisier
andauthored
build: add AMDGPU targets (#59)
* build: add AMDGPU targets To support for new ROCm chips. * rocm: drop support for old gfx They are actually not supported by ROCm anymore. Drop them until we rework the pipeline to properly link bytecodes. --------- Co-authored-by: Corentin Kerisit <corentin.kerisit@gmail.com>
1 parent b962b96 commit e1c8db3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

.github/workflows/_build.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ on:
1919
env:
2020
XLA_COMMIT: ${{ inputs.xla_commit || '9e9a0fb13b164c9875349246d1d8616a267c9bf5' }} # main from 2026-03-12
2121
ROCM_XLA_COMMIT: ${{ inputs.rocm_xla_commit || 'c40171e49b63e8a12498fae2b64f90c2406802ec' }} # rocm-jaxlib-v0.8.0 from 2025-12-30
22-
TF_ROCM_AMDGPU_TARGETS: "gfx900,gfx906,gfx908,gfx90a,gfx942,gfx1030,gfx1100"
22+
TF_ROCM_AMDGPU_TARGETS: "gfx942,gfx950,gfx1100,gfx1101,gfx1150,gfx1151,gfx1200,gfx1201"
2323
BUILDBUDDY_API_KEY: ${{ secrets.BUILDBUDDY_API_KEY }}
2424

2525
jobs:

0 commit comments

Comments
 (0)