From e905179cd1a70b11b07c631bc3218d1737969355 Mon Sep 17 00:00:00 2001
From: saienduri <saimanas.enduri@amd.com>
Date: Tue, 28 Jan 2025 02:56:32 -0800
Subject: [PATCH] docker test

---
 .github/workflows/test_gpu.yml | 20 +++++++++-----------
 1 file changed, 9 insertions(+), 11 deletions(-)

diff --git a/.github/workflows/test_gpu.yml b/.github/workflows/test_gpu.yml
index f49dc64..48e7707 100644
--- a/.github/workflows/test_gpu.yml
+++ b/.github/workflows/test_gpu.yml
@@ -30,8 +30,8 @@ jobs:
         with:
           python-version: 3.11
 
-      # Test pytorch rocm
-      - name: "pytorch rocm test"
+      # Test rocm
+      - name: "rocm test"
         run: |
           ls -a /home/runner
           ls -a /etc/podinfo
@@ -42,16 +42,14 @@ jobs:
             DEVICE_FLAG="--device /dev/dri"
           fi
           echo $DEVICE_FLAG
-          echo "TEST"
-          echo ${GHA_RENDER_DEVICES}
-          python3.11 -m venv test_venv
-          source test_venv/bin/activate
-          pip3 install --force-reinstall --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2
-          python3.11 -c "import torch; print('CUDA available:', torch.cuda.is_available())"
-          echo $GHA_RENDER_DEVICES
+          docker pull rocm/dev-ubuntu-22.04:6.3
+          docker run -dt --network=host --device=/dev/kfd --device=/dev/dri \
+            --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined \
+            --name ci_container_test \
+            rocm/dev-ubuntu-22.04:6.3
 
       # Test rocm commands
       - name: "Test rocm commands"
         run: |
-          rocm-smi
-          rocminfo
+          docker exec ci_container_test rocm-smi
+          docker exec ci_container_test rocminfo