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

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

if HAS_TRITON:

    from vllm.triton_utils.custom_cache_manager import (
        maybe_set_triton_cache_manager)
9
    from vllm.triton_utils.libentry import libentry
10

11
    __all__ += ["maybe_set_triton_cache_manager", "libentry"]