vllm/vllm/triton_utils/__init__.py

12 lines
306 B
Python

from vllm.triton_utils.importing import HAS_TRITON
__all__ = ["HAS_TRITON"]
if HAS_TRITON:
from vllm.triton_utils.custom_cache_manager import (
maybe_set_triton_cache_manager)
from vllm.triton_utils.libentry import libentry
__all__ += ["maybe_set_triton_cache_manager", "libentry"]