{{ message }}
Repository navigation
Commit 1377a2f
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

0 commit comments