Coverage for src/flag_gems/runtime/common.py: 100%

1from enum import Enum 

2 

3 

4class vendors(Enum): 

5 NVIDIA = 0 

6 CAMBRICON = 1 

7 METAX = 2 

8 ILUVATAR = 3 

9 MTHREADS = 4 

10 KUNLUNXIN = 5 

11 HYGON = 6 

12 AMD = 7 

13 AIPU = 8 

14 ASCEND = 9 

15 TSINGMICRO = 10 

16 SUNRISE = 11 

17 ENFLAME = 12 

18 

19 @classmethod 

20 def get_all_vendors(cls) -> dict: 

21 vendorDict = {} 

22 for member in cls: 

23 vendorDict[member.name.lower()] = member 

24 return vendorDict