ColossalAI/extensions/inference/__init__.py

4 lines
99 B
Python

from .inference_ops_cuda import InferenceOpsCudaExtension
__all__ = ["InferenceOpsCudaExtension"]