__init__.py

from lightx2v_platform.base.base import check_ai_device, init_ai_device
from lightx2v_platform.base.cambricon_mlu import MluDevice
from lightx2v_platform.base.dcu import DcuDevice
from lightx2v_platform.base.metax import MetaxDevice
from lightx2v_platform.base.nvidia import CudaDevice

__all__ = ["init_ai_device", "check_ai_device", "CudaDevice", "MluDevice", "MetaxDevice", "DcuDevice"]