Skip to content

Commit 20332ea

Browse files
committed
nightly: adds rocm support
1 parent 3f10099 commit 20332ea

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

.github/workflows/nightly.yaml

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,13 @@ jobs:
2323
runs_on: ["runs-on", "runner=32cpu-linux-x64"]
2424
platform: linux-amd64
2525
bazel_target: //xla/pjrt/c:pjrt_c_api_gpu_plugin
26+
- target: rocm
27+
config: "--config=rocm"
28+
artifact: libpjrt_c_api_gpu_plugin.so
29+
renamed_artifact: libpjrt_rocm.so
30+
runs_on: ["runs-on", "runner=32cpu-linux-x64", "image=ubuntu24-amd64"]
31+
platform: linux-amd64
32+
bazel_target: //xla/pjrt/c:pjrt_c_api_gpu_plugin
2633
runs-on: ubuntu-latest
2734
steps:
2835
- name: "Checking out repository"

0 commit comments

Comments
 (0)