From 9bfc6ff14d2ffcf82512f3bbd777522c07633d7e Mon Sep 17 00:00:00 2001
From: doctorpangloss <2229300+doctorpangloss@users.noreply.github.com>
Date: Tue, 21 Oct 2025 14:35:52 -0700
Subject: [PATCH] tweak dgpu

---
 .github/workflows/test.yml | 11 +++++++++++
 1 file changed, 11 insertions(+)

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: