e9f9fd3
1
2
3
4
5
6
7
8
9
10
11
12
13
14
from enum import IntEnum class DeviceId(IntEnum): GPU0 = 0, GPU1 = 1, GPU2 = 2, GPU3 = 3, GPU4 = 4, GPU5 = 5, GPU6 = 6, GPU7 = 7, XPU = 98, CPU = 99
