Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
devices.py 158 Bytes
Newer Older
PanZezhongQY's avatar
feat: cpu and cuda matmul  
PanZezhongQY committed
1
2
3
4
5 
class InfiniDeviceEnum:
    CPU = 0
    NVIDIA = 1
    CAMBRICON = 2
    ASCEND = 3
PanZezhong's avatar
fix: 删掉python enum多余逗号  
PanZezhong committed
6
7
8
9
10 
    METAX = 4
    MOORE = 5
    ILUVATAR = 6
    KUNLUN = 7
    SUGON = 8