Skip to content

Commit f9e3222

Browse files
committed
Enable ROCm support in build workflow and specify runner configuration for MI300 GPU
1 parent 12467d2 commit f9e3222

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

.github/workflows/build_wheels_linux.yml

+2-1
Original file line numberDiff line numberDiff line change
@@ -28,10 +28,11 @@ jobs:
2828
os: linux
2929
with-cpu: enable
3030
with-cuda: enable
31-
with-rocm: disable
31+
with-rocm: enable
3232
with-xpu: enable
3333
# Note: if free-threaded python is required add py3.13t here
3434
python-versions: '["3.9"]'
35+
runner: '{"rocm": "linux.rocm.gpu.mi300.2"}'
3536

3637
build:
3738
needs: generate-matrix

0 commit comments

Comments
 (0)