wehub-resource-sync
44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
# SPDX-License-Identifier: Apache-2.0
|
|
"""
|
|
Storage controllers with automatic module discovery.
|
|
|
|
All modules under this package that call ``register_store_policy`` or
|
|
``register_prefetch_policy`` at import time are discovered automatically.
|
|
To add a new policy, simply create a new module in this directory -- **no
|
|
changes to any existing file are needed**.
|
|
"""
|
|
|
|
# Standard
|
|
import importlib
|
|
import pkgutil
|
|
|
|
# First Party
|
|
from lmcache.v1.distributed.storage_controllers.eviction_controller import (
|
|
EvictionController,
|
|
L1EvictionController,
|
|
L2AdapterEvictionState,
|
|
L2EvictionController,
|
|
)
|
|
from lmcache.v1.distributed.storage_controllers.prefetch_controller import (
|
|
PrefetchController,
|
|
)
|
|
from lmcache.v1.distributed.storage_controllers.store_controller import (
|
|
StoreController,
|
|
)
|
|
|
|
# Auto-discover and import every module in this package so
|
|
# that each policy's self-registration code runs.
|
|
_PACKAGE_PATH = __path__ # type: ignore[name-defined]
|
|
_PACKAGE_NAME = __name__
|
|
for _finder, _mod_name, _is_pkg in pkgutil.iter_modules(_PACKAGE_PATH):
|
|
importlib.import_module(f".{_mod_name}", _PACKAGE_NAME)
|
|
|
|
__all__ = [
|
|
"EvictionController",
|
|
"L1EvictionController",
|
|
"L2EvictionController",
|
|
"L2AdapterEvictionState",
|
|
"PrefetchController",
|
|
"StoreController",
|
|
]
|