Skip to content

Commit

Permalink
add rocm-ci file
Browse files Browse the repository at this point in the history
  • Loading branch information
JehandadKhan committed Sep 23, 2024
1 parent 109259e commit 44a496b
Show file tree
Hide file tree
Showing 2 changed files with 20 additions and 13 deletions.
5 changes: 5 additions & 0 deletions .github/workflows/ci-build.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,7 @@ permissions:

jobs:
lint_and_typecheck:
if: false
runs-on: ubuntu-latest
timeout-minutes: 5
steps:
Expand All @@ -38,6 +39,7 @@ jobs:
- uses: pre-commit/action@2c7b3805fd2a0fd8c1884dcaebf91fc102a13ecd # ratchet: pre-commit/[email protected]

build:
if: false
name: "build ${{ matrix.name-prefix }} (py ${{ matrix.python-version }} on ${{ matrix.os }}, x64=${{ matrix.enable-x64}})"
runs-on: ${{ matrix.os }}
timeout-minutes: 60
Expand Down Expand Up @@ -103,6 +105,7 @@ jobs:
documentation:
if: false
name: Documentation - test code snippets
runs-on: ubuntu-latest
timeout-minutes: 10
Expand Down Expand Up @@ -145,6 +148,7 @@ jobs:
documentation_render:
if: false
name: Documentation - render documentation
runs-on: ubuntu-latest
timeout-minutes: 10
Expand Down Expand Up @@ -181,6 +185,7 @@ jobs:
jax2tf_test:
if: false
name: "jax2tf_test (py ${{ matrix.python-version }} on ${{ matrix.os }}, x64=${{ matrix.enable-x64}})"
runs-on: ${{ matrix.os }}
timeout-minutes: 30
Expand Down
28 changes: 15 additions & 13 deletions .github/workflows/rocm-ci.yaml
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
name: Print Environment Variables
name: ROCM CI

on: [push]

Expand All @@ -8,16 +8,18 @@ jobs:
steps:
- name: Print Environment Variables
run: printenv



- name: Set up Docker
uses: docker/setup-buildx-action@v2

- name: Try rocm-smi
run: |
docker run \
--group-add video \
$RENDER_DEVICES \
--cap-add=SYS_PTRACE \
--security-opt seccomp=unconfined \
--shm-size=64G \
rocm/rocm-terminal rocm-smi
# - name: Set up Docker
# uses: docker/setup-buildx-action@v2
#
# - name: Try rocm-smi
# run: |
# docker run \
# --group-add video \
# $RENDER_DEVICES \
# --cap-add=SYS_PTRACE \
# --security-opt seccomp=unconfined \
# --shm-size=64G \
# rocm/rocm-terminal rocm-smi

0 comments on commit 44a496b

Please sign in to comment.