gpu/cuda: differentiate V100 32GB GPU IDs
Checks
Commit Message
From: Elena Agostini <eagostini@nvidia.com>
Differentiate between GPU V100 32GB SMX2 device id
and V100 32GB PCIe device id.
Signed-off-by: Elena Agostini <eagostini@nvidia.com>
---
drivers/gpu/cuda/cuda.c | 9 +++++++--
1 file changed, 7 insertions(+), 2 deletions(-)
Comments
10/02/2022 02:58, eagostini@nvidia.com:
> From: Elena Agostini <eagostini@nvidia.com>
>
> Differentiate between GPU V100 32GB SMX2 device id
> and V100 32GB PCIe device id.
>
> Signed-off-by: Elena Agostini <eagostini@nvidia.com>
Applied, thanks.
@@ -63,7 +63,8 @@ static int cuda_driver_version;
#define NVIDIA_GPU_A30_24GB_DEVICE_ID (0x20b7)
#define NVIDIA_GPU_A10_24GB_DEVICE_ID (0x2236)
-#define NVIDIA_GPU_V100_32GB_DEVICE_ID (0x1db6)
+#define NVIDIA_GPU_V100_32GB_SXM_DEVICE_ID (0x1db5)
+#define NVIDIA_GPU_V100_32GB_PCIE_DEVICE_ID (0x1db6)
#define NVIDIA_GPU_V100_16GB_DEVICE_ID (0x1db4)
#define NVIDIA_GPU_T4_16GB_DEVICE_ID (0x1eb8)
@@ -107,7 +108,11 @@ static const struct rte_pci_id pci_id_cuda_map[] = {
},
{
RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID,
- NVIDIA_GPU_V100_32GB_DEVICE_ID)
+ NVIDIA_GPU_V100_32GB_SXM_DEVICE_ID)
+ },
+ {
+ RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID,
+ NVIDIA_GPU_V100_32GB_PCIE_DEVICE_ID)
},
{
RTE_PCI_DEVICE(NVIDIA_GPU_VENDOR_ID,