wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-10 06:10:50 +08:00
Code Issues Actions 10 Packages Projects Releases Wiki Activity

tweak dgpu

Browse Source
This commit is contained in:
doctorpangloss 2025-10-21 14:35:52 -07:00
parent 99be2c40fa
commit 9bfc6ff14d
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
.github/workflows/test.yml vendored
View File

@ -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: