Skip to content

Commit c748024

Browse files
authored
Fix lint in newer triton (#1098)
1 parent 5c7347e commit c748024

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

helion/_compat.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -152,10 +152,11 @@ def _ensure_triton_specialize_impl_alias() -> None:
152152
module: Any = triton_jit
153153
module.specialize_impl = _make_specialize_impl_wrapper() # type: ignore[assignment]
154154
return
155-
if hasattr(triton_jit, "create_specialize_impl"):
155+
create_specialize_impl = getattr(triton_jit, "create_specialize_impl", None)
156+
if create_specialize_impl is not None:
156157
module: Any = triton_jit
157158
module.specialize_impl = _make_specialize_impl_wrapper(
158-
create_factory=triton_jit.create_specialize_impl,
159+
create_factory=create_specialize_impl,
159160
) # type: ignore[assignment]
160161

161162

0 commit comments

Comments
 (0)