From 2e39e0999fe45b247c2ff4ca616e65b2b6cb35ad Mon Sep 17 00:00:00 2001 From: patientx Date: Tue, 5 Aug 2025 19:21:20 +0300 Subject: [PATCH] Update zluda.py --- comfy/customzluda/zluda.py | 20 +++++++------------- 1 file changed, 7 insertions(+), 13 deletions(-) diff --git a/comfy/customzluda/zluda.py b/comfy/customzluda/zluda.py index d54fe8d74..ff73a0b5b 100644 --- a/comfy/customzluda/zluda.py +++ b/comfy/customzluda/zluda.py @@ -7,23 +7,17 @@ os.environ.pop("HIP_HOME", None) os.environ.pop("ROCM_VERSION", None) #triton fix? -if "FLASH_ATTENTION_TRITON_AMD_ENABLE" not in os.environ: - os.environ["FLASH_ATTENTION_TRITON_AMD_ENABLE"] = "TRUE" -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)) +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) # ------------------- End ROCm/HIP Hiding ------------- # Fix for cublasLt errors on newer ZLUDA (if no hipblaslt) -if "DISABLE_ADDMM_CUDA_LT" not in os.environ: - os.environ['DISABLE_ADDMM_CUDA_LT'] = '1' +os.environ['DISABLE_ADDMM_CUDA_LT'] = '1' # ------------------- main imports ------------------- # main imports