Skip to contentSkip to conversation

gpu/cuda: differentiate V100 32GB GPU IDs

PATCH Applied [email protected] 2 participants 2 messages Feb 9, 2022 – Feb 10, 2022 DRM drivers

lore
Share & tools
lore

[email protected]

author
#0 lore raw

PATCH 1/1 gpu/cuda: differentiate V100 32GB GPU IDs

From: Elena Agostini <[email protected]>

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

Signed-off-by Elena Agostini <[email protected]>
---
 drivers/gpu/cuda/cuda.c | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)
drivers/gpu/cuda/cuda.c +7 -2
index 0ece1bb612..49c7b065a3 100644
@@ -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,
Signature · 1 line
2.17.1