Files
ColossalAI/colossalai/nn/_ops/cache_embedding/__init__.py

4 lines
132 B
Python

from .cache_mgr import CachedParamMgr
from .copyer import LimitBuffIndexCopyer
__all__ = ['CachedParamMgr', 'LimitBuffIndexCopyer']