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

Bad ideas from zluda update.

Browse Source
This commit is contained in:
Christopher Anderson 2025-08-05 06:00:55 +10:00
parent 9882dd7e99
commit 4f853403fe
1 changed files with 13 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
comfy/customzluda/zluda.py
View File

@ -7,17 +7,23 @@ os.environ.pop("HIP_HOME", None)
os.environ.pop("ROCM_VERSION", None) os.environ.pop("ROCM_VERSION", None)
#triton fix? #triton fix?
os.environ["FLASH_ATTENTION_TRITON_AMD_ENABLE"] = "TRUE" if "FLASH_ATTENTION_TRITON_AMD_ENABLE" not in os.environ:
os.environ["FLASH_ATTENTION_TRITON_AMD_AUTOTUNE"] = "TRUE" os.environ["FLASH_ATTENTION_TRITON_AMD_ENABLE"] = "TRUE"
os.environ["TRITON_DEBUG"] = "1" # Verbose logging if "FLASH_ATTENTION_TRITON_AMD_AUTOTUNE" not in os.environ:
os.environ["FLASH_ATTENTION_TRITON_AMD_AUTOTUNE"] = "TRUE"
if "TRITON_DEBUG" not in os.environ:
os.environ["TRITON_DEBUG"] = "1" # Verbose logging
paths = os.environ["PATH"].split(os.pathsep)
paths_with_rocm = [p for p in paths if "rocm" in p.lower()]
if paths_with_rocm:
print("Warning: ROCm related path(s) may need to be removed from %PATH%: " + os.pathsep.join(paths_with_rocm))
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)
# ------------------- End ROCm/HIP Hiding ------------- # ------------------- End ROCm/HIP Hiding -------------
# Fix for cublasLt errors on newer ZLUDA (if no hipblaslt) # Fix for cublasLt errors on newer ZLUDA (if no hipblaslt)
os.environ['DISABLE_ADDMM_CUDA_LT'] = '1' if "DISABLE_ADDMM_CUDA_LT" not in os.environ:
os.environ['DISABLE_ADDMM_CUDA_LT'] = '1'
# ------------------- main imports ------------------- # ------------------- main imports -------------------
# main imports # main imports