We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5c7347e commit c748024Copy full SHA for c748024
helion/_compat.py
@@ -152,10 +152,11 @@ def _ensure_triton_specialize_impl_alias() -> None:
152
module: Any = triton_jit
153
module.specialize_impl = _make_specialize_impl_wrapper() # type: ignore[assignment]
154
return
155
- if hasattr(triton_jit, "create_specialize_impl"):
+ create_specialize_impl = getattr(triton_jit, "create_specialize_impl", None)
156
+ if create_specialize_impl is not None:
157
158
module.specialize_impl = _make_specialize_impl_wrapper(
- create_factory=triton_jit.create_specialize_impl,
159
+ create_factory=create_specialize_impl,
160
) # type: ignore[assignment]
161
162
0 commit comments