This commit is contained in:
SrikanthMyakam 2024-11-11 20:37:36 +05:30 коммит произвёл LiliDeng
Родитель 9dd25511a8
Коммит 5c8d0258a6
2 изменённых файлов: 11 добавлений и 1 удалений

Просмотреть файл

@ -76,9 +76,13 @@ DEVICE_ID_DICT: Dict[str, List[str]] = {
"1016", # Mellanox Technologies MT27710 Family [ConnectX-4 Lx Virtual Function] "1016", # Mellanox Technologies MT27710 Family [ConnectX-4 Lx Virtual Function]
"1018", # Mellanox Technologies MT27800 Family [ConnectX-5 Virtual Function] "1018", # Mellanox Technologies MT27800 Family [ConnectX-5 Virtual Function]
"101a", # Mellanox Technologies MT28800 Family [ConnectX-5 Ex Virtual Function] "101a", # Mellanox Technologies MT28800 Family [ConnectX-5 Ex Virtual Function]
"101e", # Mellanox Technologies [ConnectX Family mlx5Gen Virtual Function]
"00ba", # Microsft Azure Network Adapter VF (MANA VF) "00ba", # Microsft Azure Network Adapter VF (MANA VF)
], ],
constants.DEVICE_TYPE_INFINIBAND: [
"1018", # Mellanox Technologies MT27800 Family [ConnectX-5 Virtual Function]
"101a", # Mellanox Technologies MT28800 Family [ConnectX-5 Ex Virtual Function]
"101e", # Mellanox Technologies [ConnectX Family mlx5Gen Virtual Function]
],
constants.DEVICE_TYPE_NVME: [ constants.DEVICE_TYPE_NVME: [
"b111" # Microsoft Corporation Device, Local NVMe discs "b111" # Microsoft Corporation Device, Local NVMe discs
], ],
@ -99,6 +103,9 @@ VENDOR_ID_DICT: Dict[str, List[str]] = {
VENDOR_ID_MICROSOFT, VENDOR_ID_MICROSOFT,
VENDOR_ID_MELLANOX, VENDOR_ID_MELLANOX,
], ],
constants.DEVICE_TYPE_INFINIBAND: [
VENDOR_ID_MELLANOX,
],
constants.DEVICE_TYPE_NVME: [VENDOR_ID_MICROSOFT], constants.DEVICE_TYPE_NVME: [VENDOR_ID_MICROSOFT],
constants.DEVICE_TYPE_GPU: [VENDOR_ID_NVIDIA], constants.DEVICE_TYPE_GPU: [VENDOR_ID_NVIDIA],
constants.DEVICE_TYPE_AMD_GPU: [VENDOR_ID_AMD], constants.DEVICE_TYPE_AMD_GPU: [VENDOR_ID_AMD],
@ -107,6 +114,8 @@ VENDOR_ID_DICT: Dict[str, List[str]] = {
CONTROLLER_ID_DICT: Dict[str, List[str]] = { CONTROLLER_ID_DICT: Dict[str, List[str]] = {
constants.DEVICE_TYPE_SRIOV: [ constants.DEVICE_TYPE_SRIOV: [
"0200", # Ethernet controller "0200", # Ethernet controller
],
constants.DEVICE_TYPE_INFINIBAND: [
"0207", # Infiniband controller "0207", # Infiniband controller
], ],
constants.DEVICE_TYPE_NVME: [ constants.DEVICE_TYPE_NVME: [

Просмотреть файл

@ -157,6 +157,7 @@ DATADISK_CACHING_TYPE_READONLY = "ReadOnly"
DATADISK_CACHING_TYPE_READYWRITE = "ReadWrite" DATADISK_CACHING_TYPE_READYWRITE = "ReadWrite"
DEVICE_TYPE_SRIOV = "SRIOV" DEVICE_TYPE_SRIOV = "SRIOV"
DEVICE_TYPE_INFINIBAND = "INFINIBAND"
DEVICE_TYPE_NVME = "NVME" DEVICE_TYPE_NVME = "NVME"
DEVICE_TYPE_GPU = "GPU" DEVICE_TYPE_GPU = "GPU"
DEVICE_TYPE_AMD_GPU = "AMD_GPU" DEVICE_TYPE_AMD_GPU = "AMD_GPU"