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

Update zluda.py (print Triton version)

Browse Source 
Added check for Triton version string, if it exists.
Could be useful info for troubleshooting reports.
This commit is contained in:
Rando717 2025-09-08 17:00:40 +02:00 committed by GitHub
parent b46622ffa5
commit 40199a5244
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
comfy/customzluda/zluda.py
View File

@ -431,6 +431,13 @@ try:
import triton.language as tl
print(" :: Triton core imported successfully")
# Get triton version, if possible, continue if no version string
try:
version = triton.__version__
print(f" :: Detected Triton version: {version}")
except AttributeError:
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
torch.backends.cudnn.enabled = os.environ.get("TORCH_BACKENDS_CUDNN_ENABLED", "1").strip().lower() not in {"0", "off", "false", "disable", "disabled", "no"}
if torch.backends.cudnn.enabled: