lxc: Workaround against lxc changing logfile permissions

[waydroid.git] / tools / helpers / gpu.py

diff --git a/tools/helpers/gpu.py b/tools/helpers/gpu.py
index 5f346566546fc28ad901ecdfe9ad4060a609a5b3..cf88540d11e53c133e66fd1b555aec6534daaf5f 100644 (file)
--- a/tools/helpers/gpu.py
+++ b/tools/helpers/gpu.py
@@ -2,12 +2,28 @@ import glob
 import os
 import tools.helpers.props
 
-unsupported = ["nvidia"]
+unsupported = ["nvidia", "nouveau"]
+
+def getKernelDriver(args, dev):
+    return tools.helpers.props.file_get(args, "/sys/class/drm/{}/device/uevent".format(dev), "DRIVER")
 
 def getDriNode(args):
     for node in glob.glob("/dev/dri/renderD*"):
         dev = os.path.basename(node)
-        driver = tools.helpers.props.file_get(args, "/sys/class/drm/{}/device/uevent".format(dev), "DRIVER")
-        if driver not in unsupported:
+        if getKernelDriver(args, dev) not in unsupported:
             return node
     return ""
+
+def getVulkanDriver(args, dev):
+    mapping = {
+        "i915": "intel",
+        "amdgpu": "radeon",
+        "radeon": "radeon",
+        "panfrost": "panfrost",
+        "msm": "freedreno",
+        "vc4": "broadcom",
+    }
+    kernel_driver = getKernelDriver(args, dev)
+    if kernel_driver in mapping:
+        return mapping[kernel_driver]
+    return ""