Skip to content

Commit

Permalink
Update rocm_ci.yml
Browse files Browse the repository at this point in the history
update checkout action params
  • Loading branch information
tenpercent authored Apr 1, 2024
1 parent ac2f07b commit 801003d
Showing 1 changed file with 3 additions and 5 deletions.
8 changes: 3 additions & 5 deletions .github/workflows/rocm_ci.yml
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,9 @@ jobs:
runs-on: self-hosted

steps:
- uses: actions/checkout@v2
- uses: actions/checkout@v4
with:
submodules: true
- name: Get CPU info on Ubuntu
if: contains(runner.os, 'linux')
run: |
Expand Down Expand Up @@ -47,12 +49,8 @@ jobs:
rocm-smi
rocminfo | grep "gfx"
export XFORMERS_PATH="$(mktemp -d)"
echo XFORMERS_PATH = $XFORMERS_PATH
- name: Build XFormers
run: |
git clone --recursive -b "$GIT_BRANCH" "https://github.com/$GITHUB_REPOSITORY" $XFORMERS_PATH
docker run -it --cap-add=SYS_PTRACE --security-opt seccomp=unconfined --device=/dev/kfd --device=/dev/dri --group-add video --ipc=host --shm-size 8G -v $XFORMERS_PATH:/xformers rocm/pytorch-nightly:latest
pip3 install --upgrade pip
Expand Down

0 comments on commit 801003d

Please sign in to comment.