From: Elena Agostini <eagost...@nvidia.com> Added the new Hopper 80GB GPU device ID to the list.
Signed-off-by: Elena Agostini <eagost...@nvidia.com> --- drivers/gpu/cuda/cuda.c | 12 ++++++++---- drivers/gpu/cuda/devices.h | 3 ++- 2 files changed, 10 insertions(+), 5 deletions(-) diff --git a/drivers/gpu/cuda/cuda.c b/drivers/gpu/cuda/cuda.c index 9eb748dd2e..16778e1160 100644 --- a/drivers/gpu/cuda/cuda.c +++ b/drivers/gpu/cuda/cuda.c @@ -63,10 +63,6 @@ RTE_LOG_REGISTER_DEFAULT(cuda_logtype, NOTICE); /* NVIDIA GPU address map */ static const struct rte_pci_id pci_id_cuda_map[] = { - { - RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID, - NVIDIA_GPU_H100_80GB_DEVICE_ID) - }, { RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID, NVIDIA_GPU_A40_DEVICE_ID) @@ -263,6 +259,14 @@ static const struct rte_pci_id pci_id_cuda_map[] = { RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID, NVIDIA_GPU_H100_GH3_DEVICE_ID) }, + { + RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID, + NVIDIA_GPU_H100_80GB_DEVICE_ID_1) + }, + { + RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID, + NVIDIA_GPU_H100_80GB_DEVICE_ID_2) + }, { .device_id = 0 } diff --git a/drivers/gpu/cuda/devices.h b/drivers/gpu/cuda/devices.h index 9f93132217..2968da2c8f 100644 --- a/drivers/gpu/cuda/devices.h +++ b/drivers/gpu/cuda/devices.h @@ -10,7 +10,6 @@ /* NVIDIA GPU device IDs */ -#define NVIDIA_GPU_H100_80GB_DEVICE_ID (0x2313) #define NVIDIA_GPU_A40_DEVICE_ID (0x2235) #define NVIDIA_GPU_A30_24GB_DEVICE_ID (0x20b7) #define NVIDIA_GPU_A30X_24GB_DPU_DEVICE_ID (0x20b9) @@ -71,5 +70,7 @@ #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) #endif /* CUDA_DEVICES_H */ -- 2.48.1