diff --git a/.github/workflows/fbgemm_gpu_ci_rocm.yml b/.github/workflows/fbgemm_gpu_ci_rocm.yml index 7dc36f3cd..a0f831c53 100644 --- a/.github/workflows/fbgemm_gpu_ci_rocm.yml +++ b/.github/workflows/fbgemm_gpu_ci_rocm.yml @@ -43,7 +43,7 @@ jobs: runs-on: ${{ matrix.host-machine.instance }} container: image: rocm/dev-ubuntu-22.04:${{ matrix.rocm-version }}-complete - options: --user root + options: --user root --device=/dev/kfd --device=/dev/dri --ipc=host --shm-size 16G --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined defaults: run: shell: bash