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

added rocm and hip hiding for certain nodes getting errors under zluda

Browse Source
This commit is contained in:
patientx 2025-01-19 17:49:55 +03:00 committed by GitHub
parent 88eae2b0d1
commit df418fc3fd
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
comfy/zluda.py
View File

@ -1,3 +1,15 @@
# hide rocm and hip
import os
os.environ.pop("ROCM_HOME",None)
os.environ.pop("HIP_HOME",None)
os.environ.pop("ROCM_VERSION",None)
paths=os.environ["PATH"].split(";")
paths_no_rocm=[]
for path_ in paths:
if "rocm" not in path_.lower():
paths_no_rocm.append(path_)
os.environ["PATH"]=";".join(paths_no_rocm)
import torch import torch
if torch.cuda.get_device_name().endswith("[ZLUDA]"): if torch.cuda.get_device_name().endswith("[ZLUDA]"):