# 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"]