17 lines
407 B
Python
17 lines
407 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
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)
|
|
|
|
__all__ += ["maybe_set_triton_cache_manager"]
|
|
|
|
else:
|
|
def maybe_set_triton_cache_manager() -> None:
|
|
return None
|
|
__all__ += ["maybe_set_triton_cache_manager"]
|