Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
47 changes: 47 additions & 0 deletions .github/workflows/xla_ci.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,47 @@
name: OpenXLA Internal CI

on:
push:
branches:
- 'rocm-jaxlib-v0.7.1'
- 'leo/**'
pull_request:
branches:
- 'rocm-jaxlib-v0.7.1'
- 'leo/**'
workflow_dispatch:

jobs:
build-and-test:
runs-on: linux-mi325-2
env:
# Tries to use the GitHub Repository Variable 'DOCKER_IMAGE' first.
# If not set falls back to default
DOCKER_IMAGE: ${{ vars.DOCKER_IMAGE || 'rocm/tensorflow-build@sha256:a2672ff2510b369b4a5f034272a518dc93c2e492894e3befaeef19649632ccaa' }}
CONTAINER_NAME: builder-${{ github.run_id }}
steps:
- name: Check out repository code
uses: actions/checkout@v4

- name: Start Docker container
run: |
docker run -d -t --name ${{ env.CONTAINER_NAME }} \
--device=/dev/kfd \
--device=/dev/dri \
--ipc=host \
--pid=host \
--shm-size=16G \
--group-add=video \
--cap-add=SYS_PTRACE \
--network=host \
-v ${{ github.workspace }}:/workspace \
-w /workspace \
${{ env.DOCKER_IMAGE }} \
bash

- name: Run build and test shell script
run: docker exec ${{ env.CONTAINER_NAME }} ./scripts/my-example-script.sh

- name: Teardown Docker container
if: always()
run: docker rm -f ${{ env.CONTAINER_NAME }}
Loading