1 files changed, 6 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/pci.c b/drivers/gpu/nvgpu/pci.c
index 7641c7be..e06fddd8 100644
--- a/drivers/gpu/nvgpu/pci.c
+++ b/drivers/gpu/nvgpu/pci.c
@@ -65,7 +65,12 @@ static struct gk20a_platform nvgpu_pci_device = {
 
 static struct pci_device_id nvgpu_pci_table[] = {
         {
-                PCI_DEVICE(PCI_VENDOR_ID_NVIDIA, PCI_ANY_ID),
+                PCI_DEVICE(PCI_VENDOR_ID_NVIDIA, 0x1c36),
+                .class = PCI_BASE_CLASS_DISPLAY << 16,
+                .class_mask = 0xff << 16,
+        },
+        {
+                PCI_DEVICE(PCI_VENDOR_ID_NVIDIA, 0x1c37),
                 .class = PCI_BASE_CLASS_DISPLAY << 16,
                 .class_mask = 0xff << 16,
         },