xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-05-27 17:47:52 +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>
This commit is contained in:
Agata Dobrzyniewicz 2025-09-25 12:31:14 +02:00 committed by GitHub
parent 7be9ffcd9f
commit 3c2b2ccece
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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