Skip to content

Commit b66bcb4

Browse files
adobrzynxuebwang-amd
authored andcommitted
[Bugfix] Add triton.language.tensor placeholder (vllm-project#25649)
Signed-off-by: Agata Dobrzyniewicz <adobrzyniewicz@habana.ai> Signed-off-by: xuebwang-amd <xuebwang@amd.com>
1 parent 3a3df29 commit b66bcb4

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

tests/test_triton_utils.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -69,6 +69,8 @@ def test_triton_placeholder_language():
6969
assert lang.constexpr is None
7070
assert lang.dtype is None
7171
assert lang.int64 is None
72+
assert lang.int32 is None
73+
assert lang.tensor is None
7274

7375

7476
def test_triton_placeholder_language_from_parent():

vllm/triton_utils/importing.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -93,3 +93,4 @@ def __init__(self):
9393
self.dtype = None
9494
self.int64 = None
9595
self.int32 = None
96+
self.tensor = None

0 commit comments

Comments
 (0)