Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
__init__.py 154 Bytes
Newer Older
Yang Yong (雍洋)'s avatar
Add lightx2v_platform (#541)  
Yang Yong (雍洋) committed
1
2
3
4
5 
from lightx2v_platform.base.global_var import AI_DEVICE

if AI_DEVICE == "mlu":
    from .attn.cambricon_mlu import *
    from .mm.cambricon_mlu import *