diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index 414c8fe82..32f5373d3 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -111,6 +111,17 @@ jobs:
           pylint --rcfile=.pylintrc comfy/ comfy_extras/ comfy_api/ comfy_api_nodes/
       - name: Run unit tests
         run: |
+          # always select dgpu
+          dgpu_ids=$(rocminfo | awk '/DeviceID:/ {id=$2} /Device Type:/ {if ($3 == "DISCRETE") print id}' | sort -n)
+          igpu_ids=$(rocminfo | awk '/DeviceID:/ {id=$2} /Device Type:/ {if ($3 == "INTEGRATED") print id}' | sort -n)
+          all_ids="$dgpu_ids $igpu_ids"
+          ordered_list=$(echo "${all_ids// /,} | sed 's/,$//'")
+          if [ -z "$ordered_list" ]; then
+            echo "Warning: No ROCm devices found."
+          else
+            export HIP_VISIBLE_DEVICES="$ordered_list"
+          fi
+          echo "HIP_VISIBLE_DEVICES set to: $HIP_VISIBLE_DEVICES"
           cd ${GITHUB_WORKSPACE}/src
           pytest -v tests/unit
   build_and_execute_macos: