gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Sharad Vikram
2022-12-21 17:57:33 -08:00
committed by GitHub
parent b5aafb0dab
commit 925d3d7f98
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
python/triton/language/__init__.py
View File

@@ -20,6 +20,8 @@ from .core import (
atomic_xor, atomic_xor,
bfloat16, bfloat16,
block_type, block_type,
broadcast,
broadcast_to,
cat, cat,
cdiv, cdiv,
constexpr, constexpr,
@@ -105,6 +107,8 @@ __all__ = [
"atomic_xor", "atomic_xor",
"bfloat16", "bfloat16",
"block_type", "block_type",
"broadcast",
"broadcast_to",
"builtin", "builtin",
"cat", "cat",
"cdiv", "cdiv",