From 5c8d0258a6123094311e1bb7e9d64357fee99ee3 Mon Sep 17 00:00:00 2001
From: SrikanthMyakam <m.srikanth.m@gmail.com>
Date: Mon, 11 Nov 2024 20:37:36 +0530
Subject: [PATCH] 1

---
 lisa/tools/lspci.py    | 11 ++++++++++-
 lisa/util/constants.py |  1 +
 2 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/lisa/tools/lspci.py b/lisa/tools/lspci.py
index 316414b88..2c3798a1a 100644
--- a/lisa/tools/lspci.py
+++ b/lisa/tools/lspci.py
@@ -76,9 +76,13 @@ DEVICE_ID_DICT: Dict[str, List[str]] = {
         "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_DICT: Dict[str, List[str]] = {
         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 @@ VENDOR_ID_DICT: Dict[str, List[str]] = {
 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
index 771c5317e..84caca9c2 100644
--- a/lisa/util/constants.py
+++ b/lisa/util/constants.py
@@ -157,6 +157,7 @@ DATADISK_CACHING_TYPE_READONLY = "ReadOnly"
 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"