From 65ac1464ff57860d49d583363a31f25fc8e6160e Mon Sep 17 00:00:00 2001 From: Elena Agostini Date: Wed, 19 Jan 2022 21:41:49 +0000 Subject: [PATCH] gpu/cuda: add NVIDIA GPU A100 identifier for DPU Adding a new NVIDIA GPU identifier to let driver recognize the A100 on a DPU card. Signed-off-by: Elena Agostini --- 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 d3a7234a09..fa2302621e 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.39.5