Skip to content

GitLab

Unverified Commit 3c2b2cce authored by Agata Dobrzyniewicz's avatar Agata Dobrzyniewicz Committed by GitHub
Browse files

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


Signed-off-by: default avatarAgata Dobrzyniewicz <adobrzyniewicz@habana.ai>
parent 7be9ffcd
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
tests/test_triton_utils.py
View file @ 3c2b2cce
...@@ -69,6 +69,8 @@ def test_triton_placeholder_language(): ...@@ -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():
......
vllm/triton_utils/importing.py
View file @ 3c2b2cce
...@@ -93,3 +93,4 @@ class TritonLanguagePlaceholder(types.ModuleType): ...@@ -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
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment