d/numam-dpdk
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

gpu/cuda: differentiate V100 32GB GPU IDs

Browse Source 
Differentiate between GPU V100 32GB SMX2 device id
and V100 32GB PCIe device id.

Signed-off-by: Elena Agostini <eagostini@nvidia.com>
This commit is contained in:
Elena Agostini 2022-02-10 01:58:51 +00:00 committed by Thomas Monjalon
parent d69bb47d21
commit 56b5bb509f
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
drivers/gpu/cuda/cuda.c
View File

@ -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,