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

Only use dGPU in this test

Browse Source
This commit is contained in:
doctorpangloss 2025-10-21 15:05:38 -07:00
parent 170b8348a3
commit be9856e4db
1 changed files with 2 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -111,18 +111,8 @@ 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
# only use dGPU
export HIP_VISIBLE_DEVICES=1
pytest -v tests/unit
build_and_execute_macos:
environment: "Testing"