Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
metax.py 211 Bytes
Newer Older
xuwx1's avatar
updata lightx2v  
xuwx1 committed
1
2
3
4
5
6
7 
from lightx2v_platform.base.nvidia import CudaDevice
from lightx2v_platform.registry_factory import PLATFORM_DEVICE_REGISTER


@PLATFORM_DEVICE_REGISTER("metax")
class MetaxDevice(CudaDevice):
    name = "cuda"