1

microsoft · Nov 13, 2024 · 5c8d025 · 5c8d025
1 parent 9dd2551
commit 5c8d025
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/lisa/tools/lspci.py b/lisa/tools/lspci.py
@@ -76,9 +76,13 @@
         "1016",  # Mellanox Technologies MT27710 Family [ConnectX-4 Lx Virtual Function]
         "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]
         "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: [
         "b111"  # Microsoft Corporation Device, Local NVMe discs
     ],
@@ -99,6 +103,9 @@
         VENDOR_ID_MICROSOFT,
         VENDOR_ID_MELLANOX,
     ],
+    constants.DEVICE_TYPE_INFINIBAND: [
+        VENDOR_ID_MELLANOX,
+    ],
     constants.DEVICE_TYPE_NVME: [VENDOR_ID_MICROSOFT],
     constants.DEVICE_TYPE_GPU: [VENDOR_ID_NVIDIA],
     constants.DEVICE_TYPE_AMD_GPU: [VENDOR_ID_AMD],
@@ -107,6 +114,8 @@
 CONTROLLER_ID_DICT: Dict[str, List[str]] = {
     constants.DEVICE_TYPE_SRIOV: [
         "0200",  # Ethernet controller
+    ],
+    constants.DEVICE_TYPE_INFINIBAND: [
         "0207",  # Infiniband controller
     ],
     constants.DEVICE_TYPE_NVME: [

diff --git a/lisa/util/constants.py b/lisa/util/constants.py
@@ -157,6 +157,7 @@
 DATADISK_CACHING_TYPE_READYWRITE = "ReadWrite"
 
 DEVICE_TYPE_SRIOV = "SRIOV"
+DEVICE_TYPE_INFINIBAND = "INFINIBAND"
 DEVICE_TYPE_NVME = "NVME"
 DEVICE_TYPE_GPU = "GPU"
 DEVICE_TYPE_AMD_GPU = "AMD_GPU"