From ebf86f061761325d2ab4c2233d62645d7c91dad4 Mon Sep 17 00:00:00 2001
From: Lekensteyn <lekensteyn@gmail.com>
Date: Sun, 15 Jan 2012 16:35:07 +0100
Subject: [PATCH] Detect 3D controllers as gfx device too (GH-3)

---
 bbswitch.c | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/bbswitch.c b/bbswitch.c
index 9ad81de..199229d 100644
--- a/bbswitch.c
+++ b/bbswitch.c
@@ -329,11 +329,15 @@ static int __init bbswitch_init(void) {
     struct proc_dir_entry *acpi_entry;
     struct pci_dev *pdev = NULL;
     acpi_handle igd_handle = NULL;
-    int class = PCI_CLASS_DISPLAY_VGA << 8;
 
-    while ((pdev = pci_get_class(class, pdev)) != NULL) {
+    while ((pdev = pci_get_device(PCI_ANY_ID, PCI_ANY_ID, pdev)) != NULL) {
         struct acpi_buffer buf = { ACPI_ALLOCATE_BUFFER, NULL };
         acpi_handle handle;
+        int pci_class = pdev->class >> 8;
+
+        if (pci_class != PCI_CLASS_DISPLAY_VGA &&
+            pci_class != PCI_CLASS_DISPLAY_3D)
+            continue;
 
         handle = DEVICE_ACPI_HANDLE(&pdev->dev);
         if (!handle) {