From 1c59b8d1bb6145637bd98d7cbd92d92f85d05c8a Mon Sep 17 00:00:00 2001 From: saienduri Date: Tue, 28 Jan 2025 03:02:00 -0800 Subject: [PATCH] device flag --- .github/workflows/test_gpu.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/test_gpu.yml b/.github/workflows/test_gpu.yml index 48e7707..45bd21f 100644 --- 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