diff --git a/tests/test_triton_utils.py b/tests/test_triton_utils.py index 64f72668f29c..ebb69e627e95 100644 --- a/tests/test_triton_utils.py +++ b/tests/test_triton_utils.py @@ -69,6 +69,8 @@ def test_triton_placeholder_language(): assert lang.constexpr is None assert lang.dtype is None assert lang.int64 is None + assert lang.int32 is None + assert lang.tensor is None def test_triton_placeholder_language_from_parent(): diff --git a/vllm/triton_utils/importing.py b/vllm/triton_utils/importing.py index 2a06a9b7d11e..95076a9a7c8f 100644 --- a/vllm/triton_utils/importing.py +++ b/vllm/triton_utils/importing.py @@ -93,3 +93,4 @@ def __init__(self): self.dtype = None self.int64 = None self.int32 = None + self.tensor = None