device flag

saienduri · Jan 28, 2025 · 1c59b8d · 1c59b8d
1 parent e905179
commit 1c59b8d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/test_gpu.yml b/.github/workflows/test_gpu.yml
@@ -43,7 +43,7 @@ jobs:
           fi
           echo $DEVICE_FLAG
           docker pull rocm/dev-ubuntu-22.04:6.3
-          docker run -dt --network=host --device=/dev/kfd --device=/dev/dri \
+          docker run -dt --network=host --device=/dev/kfd $DEVICE_FLAG \
             --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined \
             --name ci_container_test \
             rocm/dev-ubuntu-22.04:6.3