From 6924af73063d7f79832364e0cc79dc0b8dcff3a7 Mon Sep 17 00:00:00 2001 From: Jason Ansel Date: Fri, 7 Nov 2025 07:43:26 -0800 Subject: [PATCH] Fix lint in newer triton stack-info: PR: https://github.com/pytorch/helion/pull/1098, branch: jansel/stack/219 --- helion/_compat.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/helion/_compat.py b/helion/_compat.py index 872d27025..b7a2bae4c 100644 --- a/helion/_compat.py +++ b/helion/_compat.py @@ -152,10 +152,11 @@ def _ensure_triton_specialize_impl_alias() -> None: module: Any = triton_jit module.specialize_impl = _make_specialize_impl_wrapper() # type: ignore[assignment] return - if hasattr(triton_jit, "create_specialize_impl"): + create_specialize_impl = getattr(triton_jit, "create_specialize_impl", None) + if create_specialize_impl is not None: module: Any = triton_jit module.specialize_impl = _make_specialize_impl_wrapper( - create_factory=triton_jit.create_specialize_impl, + create_factory=create_specialize_impl, ) # type: ignore[assignment]