Temporarily disable auto enabling triton by default on AMD. (#14878)

I get freezing issues on my test machine.
This commit is contained in:
comfyanonymous 2026-07-10 18:37:59 -07:00 committed by GitHub
parent 92ddf07ba1
commit f3a36e7484
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions

View File

@ -48,7 +48,7 @@ try:
# older Triton lacks libdevice.rint on the HIP backend and hard-crashes the INT8 path.
if args.disable_triton_backend:
ck.registry.disable("triton")
elif args.enable_triton_backend or (torch.version.hip is not None and _rocm_kitchen_arch_supported()):
elif args.enable_triton_backend: # or (torch.version.hip is not None and _rocm_kitchen_arch_supported()):
try:
import triton
triton_version = tuple(int(v) for v in triton.__version__.split(".")[:2])