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

Update zluda.py

Browse Source
This commit is contained in:
patientx 2025-05-13 18:57:09 +03:00 committed by GitHub
parent a96e17fa12
commit 3609a0cf35
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
comfy/customzluda/zluda.py
View File

@ -6,6 +6,11 @@ os.environ.pop("ROCM_HOME", None)
os.environ.pop("HIP_HOME", None)
os.environ.pop("ROCM_VERSION", None)
#triton fix?
os.environ["FLASH_ATTENTION_TRITON_AMD_ENABLE"] = "TRUE"
os.environ["FLASH_ATTENTION_TRITON_AMD_AUTOTUNE"] = "TRUE"
os.environ["TRITON_DEBUG"] = "1" # Verbose logging
paths = os.environ["PATH"].split(";")
paths_no_rocm = [p for p in paths if "rocm" not in p.lower()]
os.environ["PATH"] = ";".join(paths_no_rocm)
@ -17,6 +22,10 @@ os.environ['DISABLE_ADDMM_CUDA_LT'] = '1'
# ------------------- main imports -------------------
# main imports
import torch
torch._dynamo.config.suppress_errors = True # Skip compilation errors
torch._dynamo.config.optimize_ddp = False # Disable distributed optimizations
import ctypes
import shutil
import subprocess