[FRONTEND] Export broadcast and broadcast_to in triton.language (#1007)

2022-12-21 17:57:33 -08:00
parent b5aafb0dab
commit 925d3d7f98
1 changed files with 4 additions and 0 deletions
--- a/python/triton/language/init.py
+++ b/python/triton/language/init.py
@@ -20,6 +20,8 @@ from .core import (
    atomic_xor,
    bfloat16,
    block_type,
+    broadcast,
+    broadcast_to,
    cat,
    cdiv,
    constexpr,
@@ -105,6 +107,8 @@ __all__ = [
    "atomic_xor",
    "bfloat16",
    "block_type",
+    "broadcast",
+    "broadcast_to",
    "builtin",
    "cat",
    "cdiv",