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

Reply via email to