From: Elena Agostini <[email protected]> More gpu devices for h100 and l40
Signed-off-by: Elena Agostini <[email protected]> --- drivers/gpu/cuda/cuda.c | 66 ++++++++++++++++++++++++++++++++++---- drivers/gpu/cuda/devices.h | 23 ++++++++++--- 2 files changed, 78 insertions(+), 11 deletions(-) diff --git a/drivers/gpu/cuda/cuda.c b/drivers/gpu/cuda/cuda.c index 16778e1160..a94c4baa64 100644 --- a/drivers/gpu/cuda/cuda.c +++ b/drivers/gpu/cuda/cuda.c @@ -87,6 +87,10 @@ static const struct rte_pci_id pci_id_cuda_map[] = { RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID, NVIDIA_GPU_A10M_DEVICE_ID) }, + { + RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID, + NVIDIA_GPU_A800_DEVICE_ID) + }, { RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID, NVIDIA_GPU_A100_40GB_SXM4_DEVICE_ID) @@ -121,11 +125,23 @@ static const struct rte_pci_id pci_id_cuda_map[] = { }, { RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID, - NVIDIA_GPU_GA100_80GB_ROYB_EB1_DEVICE_ID) + NVIDIA_GPU_A800_80GB_SXM_DEVICE_ID) + }, + { + RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID, + NVIDIA_GPU_A800_80GB_DEVICE_ID) + }, + { + RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID, + NVIDIA_GPU_A800_40GB_DEVICE_ID) + }, + { + RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID, + NVIDIA_GPU_A800_80GB_ROYB_DEVICE_ID) }, { RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID, - NVIDIA_GPU_GA100_80GB_ROYB_DEVICE_ID) + NVIDIA_GPU_GA100_80GB_ROYB_DEVICE_ID) }, { RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID, @@ -249,23 +265,59 @@ static const struct rte_pci_id pci_id_cuda_map[] = { }, { RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID, - NVIDIA_GPU_HOPPER_GRACE) + NVIDIA_GPU_H100_GH1_DEVICE_ID) }, { RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID, - NVIDIA_GPU_H100_GH1_DEVICE_ID) + NVIDIA_GPU_H100_GH3_DEVICE_ID) }, { RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID, - NVIDIA_GPU_H100_GH3_DEVICE_ID) + NVIDIA_GPU_H100L_DEVICE_ID) + }, + { + RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID, + NVIDIA_GPU_H100XM_DEVICE_ID) + }, + { + RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID, + NVIDIA_GPU_H100_PCIE_DEVICE_ID) + }, + { + RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID, + NVIDIA_GPU_H100_GENERIC_DEVICE_ID) + }, + { + RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID, + NVIDIA_GPU_H100_80GB_DEVICE_ID) + }, + { + RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID, + NVIDIA_GPU_H800_PCIE_DEVICE_ID) + }, + { + RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID, + NVIDIA_GPU_H800_DEVICE_ID) + }, + { + RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID, + NVIDIA_GPU_H800L_DEVICE_ID) + }, + { + RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID, + NVIDIA_GPU_L40_DEVICE_ID) + }, + { + RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID, + NVIDIA_GPU_L40S_DEVICE_ID) }, { RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID, - NVIDIA_GPU_H100_80GB_DEVICE_ID_1) + NVIDIA_GPU_L40G_DEVICE_ID) }, { RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID, - NVIDIA_GPU_H100_80GB_DEVICE_ID_2) + NVIDIA_GPU_L4_DEVICE_ID) }, { .device_id = 0 diff --git a/drivers/gpu/cuda/devices.h b/drivers/gpu/cuda/devices.h index 2968da2c8f..9b5299f3f1 100644 --- a/drivers/gpu/cuda/devices.h +++ b/drivers/gpu/cuda/devices.h @@ -16,6 +16,7 @@ #define NVIDIA_GPU_A10_24GB_DEVICE_ID (0x2236) #define NVIDIA_GPU_A10G_DEVICE_ID (0x2237) #define NVIDIA_GPU_A10M_DEVICE_ID (0x2238) +#define NVIDIA_GPU_A800_DEVICE_ID (0x2238) #define NVIDIA_GPU_A100_40GB_SXM4_DEVICE_ID (0x20b0) #define NVIDIA_GPU_A100_40GB_PCIE_DEVICE_ID (0x20b1) @@ -26,7 +27,10 @@ #define NVIDIA_GPU_GA100_PG506_207 (0x20f0) #define NVIDIA_GPU_GA100_PCIE (0x20f1) #define NVIDIA_GPU_GA100_PG506_217 (0x20f2) -#define NVIDIA_GPU_GA100_80GB_ROYB_EB1_DEVICE_ID (0x20f5) +#define NVIDIA_GPU_A800_80GB_SXM_DEVICE_ID (0x20f3) +#define NVIDIA_GPU_A800_80GB_DEVICE_ID (0x20f5) +#define NVIDIA_GPU_A800_40GB_DEVICE_ID (0x20f6) +#define NVIDIA_GPU_A800_80GB_ROYB_DEVICE_ID (0x20bd) #define NVIDIA_GPU_GA100_80GB_ROYB_DEVICE_ID (0x20fd) #define NVIDIA_GPU_V100_16GB_SXM2_DEVICE_ID (0x1db1) @@ -67,10 +71,21 @@ #define NVIDIA_GPU_T4G (0x1eb4) #define NVIDIA_GPU_T4 (0x1eb8) -#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) +#define NVIDIA_GPU_H100L_DEVICE_ID (0x2321) +#define NVIDIA_GPU_H100XM_DEVICE_ID (0x2330) +#define NVIDIA_GPU_H100_PCIE_DEVICE_ID (0x2331) +#define NVIDIA_GPU_H100_GENERIC_DEVICE_ID (0x2339) +#define NVIDIA_GPU_H100_80GB_DEVICE_ID (0x2313) +#define NVIDIA_GPU_H800_PCIE_DEVICE_ID (0x2322) +#define NVIDIA_GPU_H800_DEVICE_ID (0x2324) +#define NVIDIA_GPU_H800L_DEVICE_ID (0x233a) + +#define NVIDIA_GPU_L40_DEVICE_ID (0x26b5) +#define NVIDIA_GPU_L40S_DEVICE_ID (0x26b9) +#define NVIDIA_GPU_L40G_DEVICE_ID (0x26b8) + +#define NVIDIA_GPU_L4_DEVICE_ID (0x27b8) #endif /* CUDA_DEVICES_H */ -- 2.48.1
