Merge remote-tracking branch 'origin/master' into phil/fused-attention-perf-fixup

2023-01-03 18:31:34 -08:00
parent 737e43a627 8460ea3df1
commit 8df1fa5e5b
7 changed files with 67 additions and 7 deletions
--- 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",