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

Update zluda.py (cleanup print Triton version)

Browse Source 
compacted, without exception, silent if no version string
This commit is contained in:
Rando717 2025-09-09 19:30:54 +02:00 committed by GitHub
parent ce8900fa25
commit 13ba6a8a8d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
comfy/customzluda/zluda.py
View File

@ -447,12 +447,11 @@ try:
import triton.language as tl import triton.language as tl
print(" :: Triton core imported successfully") print(" :: Triton core imported successfully")
# Get triton version, if possible, continue if no version string # Detect Triton version, if possible
try: version = getattr(triton, "__version__", None)
version = triton.__version__ if version:
print(f" :: Detected Triton version: {version}") print(f" :: Detected Triton version: {version}")
except AttributeError: # else: do nothing
print(" :: Detected Triton version: no info")
# This needs to be up here, so it can disable cudnn before anything can even think about using it # This needs to be up here, so it can disable cudnn before anything can even think about using it
torch.backends.cudnn.enabled = os.environ.get("TORCH_BACKENDS_CUDNN_ENABLED", "1").strip().lower() not in {"0", "off", "false", "disable", "disabled", "no"} torch.backends.cudnn.enabled = os.environ.get("TORCH_BACKENDS_CUDNN_ENABLED", "1").strip().lower() not in {"0", "off", "false", "disable", "disabled", "no"}