[Frontend] Fix import for libdevice (#1028)

This is a hotfix for issue 1 in
https://github.com/openai/triton/issues/1017
This commit is contained in:
Keren Zhou
2023-01-03 18:48:05 -05:00
committed by GitHub
parent 678b9f53a2
commit 8460ea3df1

View File

@@ -5,6 +5,7 @@ from ..impl import (
ir, ir,
builtin, builtin,
) )
from . import libdevice
from .core import ( from .core import (
abs, abs,
arange, arange,
@@ -130,6 +131,7 @@ __all__ = [
"int64", "int64",
"int8", "int8",
"ir", "ir",
"libdevice",
"load", "load",
"log", "log",
"max", "max",