diff --git a/.github/workflows/rocm_ci.yml b/.github/workflows/rocm_ci.yml index 0dc8d1cefd..af85cc776b 100644 --- a/.github/workflows/rocm_ci.yml +++ b/.github/workflows/rocm_ci.yml @@ -11,8 +11,8 @@ on: jobs: build: - if: github.repository == 'rocm/xformers' - runs-on: self-hosted + if: github.repository == 'rocm/xformers' || github.repository == 'facebookresearch/xformers' + runs-on: rocm container: image: 'rocm/pytorch-nightly:latest' options: ' --cap-add=SYS_PTRACE --security-opt seccomp=unconfined --device=/dev/kfd --device=/dev/dri --group-add video --ipc=host --shm-size 8G '