diff --git a/python/triton/language/__init__.py b/python/triton/language/__init__.py index 4618794a0..0e7875e42 100644 --- 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",