From 8460ea3df1a2aac65e0e717793bbbfb11d227663 Mon Sep 17 00:00:00 2001 From: Keren Zhou Date: Tue, 3 Jan 2023 18:48:05 -0500 Subject: [PATCH] [Frontend] Fix import for libdevice (#1028) This is a hotfix for issue 1 in https://github.com/openai/triton/issues/1017 --- python/triton/language/__init__.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/python/triton/language/__init__.py b/python/triton/language/__init__.py index 0e7875e42..038e26bbe 100644 --- a/python/triton/language/__init__.py +++ b/python/triton/language/__init__.py @@ -5,6 +5,7 @@ from ..impl import ( ir, builtin, ) +from . import libdevice from .core import ( abs, arange, @@ -130,6 +131,7 @@ __all__ = [ "int64", "int8", "ir", + "libdevice", "load", "log", "max",