Only auto-enable the ROCm comfy-kitchen Triton backend on matrix-core… · code/app-python-comfyui@1377a2f · GitHub
Skip to content

Commit 1377a2f

Browse files
authored
Only auto-enable the ROCm comfy-kitchen Triton backend on matrix-core GPUs (Comfy-Org#14869)
Comfy-Org#14862 auto-enables the comfy-kitchen Triton backend whenever torch.version.hip is set and Triton >= 3.7. The INT8 matmul kernels compile tl.dot to matrix-core instructions (WMMA on RDNA3+/gfx11xx-gfx12xx, MFMA on CDNA/gfx9xx); RDNA1/RDNA2 (gfx10xx) have neither, so the auto-enabled INT8 path hangs the GPU there (reported on RDNA2 + triton-windows 3.7.1: native and custom-node INT8 freeze until reset). Gate the automatic ROCm default on GPU architecture as well as Triton version so RDNA1/RDNA2 stay on the working eager fallback. Add --disable-triton-backend as an explicit override; --enable-triton-backend still force-enables on any arch.
1 parent 206b924 commit 1377a2f

2 files changed

Lines changed: 23 additions & 2 deletions

File tree

‎comfy/cli_args.py‎

Lines changed: 1 addition & 0 deletions

‎comfy/quant_ops.py‎

Lines changed: 22 additions & 2 deletions

0 commit comments

Comments
 (0)