Skip to contentSkip to conversation

gpu/cuda: add NVIDIA GPU A100 identifier for DPU

PATCH Applied [email protected] 2 participants 2 messages Jan 19, 2022 – Jan 20, 2022 DRM drivers

lore
Share & tools
lore

[email protected]

author
#0 lore raw

PATCH 1/1 gpu/cuda: add NVIDIA GPU A100 identifier for DPU

From: Elena Agostini <[email protected]>

Adding a new NVIDIA GPU identifier to let
driver recognize the A100 on a DPU card.

Signed-off-by Elena Agostini <[email protected]>
---
 drivers/gpu/cuda/cuda.c | 5 +++++
 1 file changed, 5 insertions(+)
drivers/gpu/cuda/cuda.c +5 -0
index 882df08e56..c295e7cc70 100644
@@ -58,6 +58,7 @@ static int cuda_driver_version;
 /* NVIDIA GPU device IDs */
 #define NVIDIA_GPU_A100_40GB_DEVICE_ID (0x20f1)
 #define NVIDIA_GPU_A100_80GB_DEVICE_ID (0x20b5)
+#define NVIDIA_GPU_A100_80GB_DPU_DEVICE_ID (0x20b8)
 
 #define NVIDIA_GPU_A30_24GB_DEVICE_ID (0x20b7)
 #define NVIDIA_GPU_A10_24GB_DEVICE_ID (0x2236)

@@ -92,6 +93,10 @@ static const struct rte_pci_id pci_id_cuda_map[] = {
 		RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID,
 				NVIDIA_GPU_A100_80GB_DEVICE_ID)
 	},
+	{
+		RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID,
+				NVIDIA_GPU_A100_80GB_DPU_DEVICE_ID)
+	},
 	{
 		RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID,
 				NVIDIA_GPU_A30_24GB_DEVICE_ID)
Signature · 1 line
2.17.1