Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 17:24:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

Set has_triton to True for CPU backend

Browse Source
This commit is contained in:
Disty0
2026-06-21 14:25:16 +03:00
parent e4c53cc743
commit da3907a17c
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/devices.py
+8 -2
View File
@@ -401,8 +401,14 @@ def test_triton(early: bool = False):
return triton_ok
t0 = time.time()
try:
from torch.utils._triton import has_triton as torch_has_triton
if torch_has_triton():
if torch._dynamo.config.disable: # pylint: disable=protected-access
triton_is_available = False
elif backend == "cpu": # CPUs can use torch.compile / Inductor without Triton
triton_is_available = True
else:
from torch.utils._triton import has_triton as torch_has_triton
triton_is_available = torch_has_triton()
if triton_is_available:
if early:
return True
def test_triton_func(a,b,c):