[PATCH v1] gpu/cuda: add NVIDIA GPU A100 identifier for DPU
eagostini at nvidia.com
eagostini at nvidia.com
Wed Jan 19 22:41:49 CET 2022
From: Elena Agostini <eagostini at nvidia.com>
Adding a new NVIDIA GPU identifier to let
driver recognize the A100 on a DPU card.
Signed-off-by: Elena Agostini <eagostini at nvidia.com>
---
drivers/gpu/cuda/cuda.c | 5 +++++
1 file changed, 5 insertions(+)
diff --git a/drivers/gpu/cuda/cuda.c b/drivers/gpu/cuda/cuda.c
index 882df08e56..c295e7cc70 100644
--- a/drivers/gpu/cuda/cuda.c
+++ b/drivers/gpu/cuda/cuda.c
@@ -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)
--
2.17.1
More information about the dev
mailing list