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

Actually enable IPEX XPU support

Browse Source
This commit is contained in:
doctorpangloss 2024-07-05 13:10:31 -07:00
parent 7c6d716541
commit 7a20b189a5
2 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -19,7 +19,7 @@ jobs:
- labels: [self-hosted, Linux, X64, cpu]
container: "ubuntu"
- labels: [self-hosted, Linux, X64, ipex-a380-6gb]
container: "intel/intel-optimized-pytorch:2.3.0-pip-base"
container: "intel/intel-optimized-pytorch:2.1.20-xpu-pip-base"
- labels: [self-hosted, Linux, X64, rocm-7600-8gb]
container: "rocm/pytorch:rocm6.0.2_ubuntu22.04_py3.10_pytorch_2.1.2"
- labels: [self-hosted, Linux, X64, cuda-3060-12gb]

1
tests/conftest.py
View File

@ -54,7 +54,6 @@ def has_gpu() -> bool:
try:
import intel_extension_for_pytorch as ipex
has_gpu = ipex.xpu.device_count() > 0
return has_gpu
except ImportError:
try:
import torch