Skip to content

Commit d66e5c3

Browse files
authored
Temporarily move triton_key import to inner function to unblock older torch versions (#395)
1 parent 8295250 commit d66e5c3

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

helion/autotuner/base_cache.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,6 @@
1111

1212
from torch._inductor.codecache import build_code_hash
1313
from torch._inductor.codecache import torch_key
14-
from torch._inductor.runtime.triton_compat import triton_key
1514

1615
from .._utils import counters
1716

@@ -39,6 +38,8 @@ def torch_key_wrapper() -> str:
3938

4039
@functools.cache
4140
def triton_key_wrapper() -> str:
41+
from torch._inductor.runtime.triton_compat import triton_key
42+
4243
return triton_key()
4344

4445

0 commit comments

Comments
 (0)