__init__.py 281 Bytes
Newer Older
1
2
# SPDX-License-Identifier: Apache-2.0

3
from vllm.triton_utils.importing import HAS_TRITON
4

5
6
7
8
9
10
11
__all__ = ["HAS_TRITON"]

if HAS_TRITON:

    from vllm.triton_utils.custom_cache_manager import (
        maybe_set_triton_cache_manager)

12
    __all__ += ["maybe_set_triton_cache_manager"]