Skip to content

GitLab

"alembic.ini" did not exist on "394e7a416603d4f9b60e60554ac5194c58f3b359"
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