refactor: support pci class prefixes

Signed-off-by: thxCode <thxcode0824@gmail.com>

Author: thxCode

gpustack_runtime/detector/__utils__.py

@@ -13,6 +13,8 @@
 from pathlib import Path
 from typing import Any
 
+from gpustack_runtime import envs
+
 
 @dataclass
 class PCIDevice:
@@ -69,7 +71,10 @@ def get_pci_devices(
         return pci_devices
 
     if not class_prefix:
-        class_prefix = ["0x02", "0x03", "0x12", "0x0b"]
+        class_prefix = []
+        for prefix in envs.GPUSTACK_RUNTIME_DETECT_PCI_CLASS_PREFIXES:
+            if prefix:
+                class_prefix.append(f"0x{prefix.lower()}")
 
     if address and isinstance(address, str):
         address = [address]

gpustack_runtime/envs.py

@@ -39,6 +39,11 @@
     Set true to disable PCI check during detection.
     Useful for WSL environments, where PCI information may not be available.
     """
+    GPUSTACK_RUNTIME_DETECT_PCI_CLASS_PREFIXES: set[str] | None = None
+    """
+    The PCI class prefixes to check during detection.
+    When `GPUSTACK_RUNTIME_DETECT_NO_PCI_CHECK` is false, the detector will check if the PCI class of the device starts with any of the given prefixes.
+    """
     GPUSTACK_RUNTIME_DETECT_NO_TOOLKIT_CALL: bool = False
     """
     Set true to disable toolkit calls during detection.
@@ -374,6 +379,16 @@
         lambda: to_bool(getenv("GPUSTACK_RUNTIME_DETECT_NO_PCI_CHECK")),
         lambda: any(x in get_os_release() for x in ("microsoft", "wsl")),
     ),
+    "GPUSTACK_RUNTIME_DETECT_PCI_CLASS_PREFIXES": lambda: to_set(
+        getenvs(
+            [
+                "GPUSTACK_RUNTIME_DETECT_PCI_CLASS_PREFIXES",
+                "GPUSTACK_PCI_CLASS_PREFIXES",
+            ],
+            "02,03,0b,12",
+        ),
+        sep=",",
+    ),
     "GPUSTACK_RUNTIME_DETECT_NO_TOOLKIT_CALL": lambda: to_bool(
         getenv(
             "GPUSTACK_RUNTIME_DETECT_NO_TOOLKIT_CALL",