From 925d3d7f98bac449a775cf183f173bb5168418d9 Mon Sep 17 00:00:00 2001 From: Sharad Vikram Date: Wed, 21 Dec 2022 17:57:33 -0800 Subject: [PATCH] [FRONTEND] Export `broadcast` and `broadcast_to` in `triton.language` (#1007) --- python/triton/language/__init__.py | 4 ++++ 1 file changed, 4 insertions(+) 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",