From: Elena Agostini <eagost...@nvidia.com> Adding a new NVIDIA GPU identifier to let driver recognize the A100 on a DPU card.
Signed-off-by: Elena Agostini <eagost...@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