[PATCH 6/9] gpu/cuda: more gpu ID for h100 and l40

[Gregory Etelson]

From: Elena Agostini <eagostini at nvidia.com>

More gpu devices for h100 and l40

Signed-off-by: Elena Agostini <eagostini at nvidia.com>
---
 drivers/gpu/cuda/cuda.c    | 66 ++++++++++++++++++++++++++++++++++----
 drivers/gpu/cuda/devices.h | 23 ++++++++++---
 2 files changed, 78 insertions(+), 11 deletions(-)

diff --git a/drivers/gpu/cuda/cuda.c b/drivers/gpu/cuda/cuda.c
index 16778e1160..a94c4baa64 100644
--- a/drivers/gpu/cuda/cuda.c
+++ b/drivers/gpu/cuda/cuda.c
@@ -87,6 +87,10 @@ static const struct rte_pci_id pci_id_cuda_map[] = {
 		RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID,
 				NVIDIA_GPU_A10M_DEVICE_ID)
 	},
+	{
+		RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID,
+				NVIDIA_GPU_A800_DEVICE_ID)
+	},
 	{
 		RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID,
 				NVIDIA_GPU_A100_40GB_SXM4_DEVICE_ID)
@@ -121,11 +125,23 @@ static const struct rte_pci_id pci_id_cuda_map[] = {
 	},
 	{
 		RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID,
-				NVIDIA_GPU_GA100_80GB_ROYB_EB1_DEVICE_ID)
+				NVIDIA_GPU_A800_80GB_SXM_DEVICE_ID)
+	},
+	{
+		RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID,
+			NVIDIA_GPU_A800_80GB_DEVICE_ID)
+	},
+	{
+		RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID,
+			NVIDIA_GPU_A800_40GB_DEVICE_ID)
+	},
+	{
+		RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID,
+			NVIDIA_GPU_A800_80GB_ROYB_DEVICE_ID)
 	},
 	{
 		RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID,
-				NVIDIA_GPU_GA100_80GB_ROYB_DEVICE_ID)
+			NVIDIA_GPU_GA100_80GB_ROYB_DEVICE_ID)
 	},
 	{
 		RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID,
@@ -249,23 +265,59 @@ static const struct rte_pci_id pci_id_cuda_map[] = {
 	},
 	{
 		RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID,
-				NVIDIA_GPU_HOPPER_GRACE)
+				NVIDIA_GPU_H100_GH1_DEVICE_ID)
 	},
 	{
 		RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID,
-				NVIDIA_GPU_H100_GH1_DEVICE_ID)
+				NVIDIA_GPU_H100_GH3_DEVICE_ID)
 	},
 	{
 		RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID,
-				NVIDIA_GPU_H100_GH3_DEVICE_ID)
+				NVIDIA_GPU_H100L_DEVICE_ID)
+	},
+	{
+		RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID,
+				NVIDIA_GPU_H100XM_DEVICE_ID)
+	},
+	{
+		RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID,
+			NVIDIA_GPU_H100_PCIE_DEVICE_ID)
+	},
+	{
+		RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID,
+			NVIDIA_GPU_H100_GENERIC_DEVICE_ID)
+	},
+	{
+		RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID,
+			NVIDIA_GPU_H100_80GB_DEVICE_ID)
+	},
+	{
+		RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID,
+			NVIDIA_GPU_H800_PCIE_DEVICE_ID)
+	},
+	{
+		RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID,
+			NVIDIA_GPU_H800_DEVICE_ID)
+	},
+	{
+		RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID,
+			NVIDIA_GPU_H800L_DEVICE_ID)
+	},
+	{
+		RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID,
+			NVIDIA_GPU_L40_DEVICE_ID)
+	},
+	{
+		RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID,
+			NVIDIA_GPU_L40S_DEVICE_ID)
 	},
 	{
 		RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID,
-				NVIDIA_GPU_H100_80GB_DEVICE_ID_1)
+			NVIDIA_GPU_L40G_DEVICE_ID)
 	},
 	{
 		RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID,
-				NVIDIA_GPU_H100_80GB_DEVICE_ID_2)
+			NVIDIA_GPU_L4_DEVICE_ID)
 	},
 	{
 		.device_id = 0
diff --git a/drivers/gpu/cuda/devices.h b/drivers/gpu/cuda/devices.h
index 2968da2c8f..9b5299f3f1 100644
--- a/drivers/gpu/cuda/devices.h
+++ b/drivers/gpu/cuda/devices.h
@@ -16,6 +16,7 @@
 #define NVIDIA_GPU_A10_24GB_DEVICE_ID (0x2236)
 #define NVIDIA_GPU_A10G_DEVICE_ID (0x2237)
 #define NVIDIA_GPU_A10M_DEVICE_ID (0x2238)
+#define NVIDIA_GPU_A800_DEVICE_ID (0x2238)
 
 #define NVIDIA_GPU_A100_40GB_SXM4_DEVICE_ID (0x20b0)
 #define NVIDIA_GPU_A100_40GB_PCIE_DEVICE_ID (0x20b1)
@@ -26,7 +27,10 @@
 #define NVIDIA_GPU_GA100_PG506_207 (0x20f0)
 #define NVIDIA_GPU_GA100_PCIE (0x20f1)
 #define NVIDIA_GPU_GA100_PG506_217 (0x20f2)
-#define NVIDIA_GPU_GA100_80GB_ROYB_EB1_DEVICE_ID (0x20f5)
+#define NVIDIA_GPU_A800_80GB_SXM_DEVICE_ID (0x20f3)
+#define NVIDIA_GPU_A800_80GB_DEVICE_ID (0x20f5)
+#define NVIDIA_GPU_A800_40GB_DEVICE_ID (0x20f6)
+#define NVIDIA_GPU_A800_80GB_ROYB_DEVICE_ID (0x20bd)
 #define NVIDIA_GPU_GA100_80GB_ROYB_DEVICE_ID (0x20fd)
 
 #define NVIDIA_GPU_V100_16GB_SXM2_DEVICE_ID (0x1db1)
@@ -67,10 +71,21 @@
 #define NVIDIA_GPU_T4G (0x1eb4)
 #define NVIDIA_GPU_T4 (0x1eb8)
 
-#define NVIDIA_GPU_HOPPER_GRACE (0x2342)
 #define NVIDIA_GPU_H100_GH1_DEVICE_ID (0x2342)
 #define NVIDIA_GPU_H100_GH3_DEVICE_ID (0x237e)
-#define NVIDIA_GPU_H100_80GB_DEVICE_ID_1 (0x2331)
-#define NVIDIA_GPU_H100_80GB_DEVICE_ID_2 (0x2313)
+#define NVIDIA_GPU_H100L_DEVICE_ID (0x2321)
+#define NVIDIA_GPU_H100XM_DEVICE_ID (0x2330)
+#define NVIDIA_GPU_H100_PCIE_DEVICE_ID (0x2331)
+#define NVIDIA_GPU_H100_GENERIC_DEVICE_ID (0x2339)
+#define NVIDIA_GPU_H100_80GB_DEVICE_ID (0x2313)
 
+#define NVIDIA_GPU_H800_PCIE_DEVICE_ID (0x2322)
+#define NVIDIA_GPU_H800_DEVICE_ID (0x2324)
+#define NVIDIA_GPU_H800L_DEVICE_ID (0x233a)
+
+#define NVIDIA_GPU_L40_DEVICE_ID (0x26b5)
+#define NVIDIA_GPU_L40S_DEVICE_ID (0x26b9)
+#define NVIDIA_GPU_L40G_DEVICE_ID (0x26b8)
+
+#define NVIDIA_GPU_L4_DEVICE_ID (0x27b8)
 #endif /* CUDA_DEVICES_H */
-- 
2.48.1