xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-06-04 21:15:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Bugfix] Add triton.language.tensor placeholder (#25649)

Browse Source 
Signed-off-by: Agata Dobrzyniewicz <adobrzyniewicz@habana.ai>
Signed-off-by: yewentao256 <zhyanwentao@126.com>
This commit is contained in:
Agata Dobrzyniewicz 2025-09-25 12:31:14 +02:00 committed by yewentao256
parent fd28c58825
commit 034c0152db
2 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_triton_utils.py
View File

@ -69,6 +69,8 @@ def test_triton_placeholder_language():
assert lang.constexpr is None assert lang.constexpr is None
assert lang.dtype is None assert lang.dtype is None
assert lang.int64 is None assert lang.int64 is None
assert lang.int32 is None
assert lang.tensor is None
def test_triton_placeholder_language_from_parent(): def test_triton_placeholder_language_from_parent():

1
vllm/triton_utils/importing.py
View File

@ -93,3 +93,4 @@ class TritonLanguagePlaceholder(types.ModuleType):
self.dtype = None self.dtype = None
self.int64 = None self.int64 = None
self.int32 = None self.int32 = None
self.tensor = None