Skip to content

Commit 15d494a

Browse files
authored
Build CPU plugin for Linux ARM64 (#57)
* Build CPU plugin for Linux ARM64 * wip
1 parent 181f69d commit 15d494a

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

.github/workflows/_build.yaml

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -65,6 +65,13 @@ jobs:
6565
runs_on: ["runs-on", "runner=32cpu-linux-x64", "image=ubuntu22-amd64"]
6666
platform: linux-amd64
6767
bazel_target: //xla/pjrt/c:pjrt_c_api_cpu_plugin
68+
- target: cpu
69+
config: "--config=hermetic_linux_arm64 --config=remote --config=bzlmod --remote_header=x-buildbuddy-api-key=$BUILDBUDDY_API_KEY --platforms @toolchains_llvm_bootstrapped//platforms:linux_arm64"
70+
artifact: bazel-bin/xla/pjrt/c/libpjrt_c_api_cpu_plugin.so
71+
renamed_artifact: libpjrt_cpu.so
72+
runs_on: ["runs-on", "runner=32cpu-linux-x64", "image=ubuntu22-amd64"]
73+
platform: linux-arm64
74+
bazel_target: //xla/pjrt/c:pjrt_c_api_cpu_plugin
6875
steps:
6976
- uses: runs-on/action@v1
7077
if: matrix.pjrt.platform != 'darwin-arm64' && matrix.pjrt.platform != 'darwin-amd64'

0 commit comments

Comments
 (0)