__init__.py 242 Bytes
Newer Older
1
from vllm.triton_utils.importing import HAS_TRITON
2

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

if HAS_TRITON:

    from vllm.triton_utils.custom_cache_manager import (
        maybe_set_triton_cache_manager)

    __all__ += ["maybe_set_triton_cache_manager"]