tools/helpers/gpu.py

   1 import glob
   2 import os
   3 import tools.helpers.props
   4
   5 unsupported = ["nvidia", "nouveau"]
   6
   7 def getKernelDriver(args, dev):
   8     return tools.helpers.props.file_get(args, "/sys/class/drm/{}/device/uevent".format(dev), "DRIVER")
   9
  10 def getDriNode(args):
  11     for node in glob.glob("/dev/dri/renderD*"):
  12         dev = os.path.basename(node)
  13         if getKernelDriver(args, dev) not in unsupported:
  14             return node
  15     return ""
  16
  17 def getVulkanDriver(args, dev):
  18     mapping = {
  19         "i915": "intel",
  20         "amdgpu": "radeon",
  21         "radeon": "radeon",
  22         "panfrost": "panfrost",
  23         "msm": "freedreno",
  24         "vc4": "broadcom",
  25     }
  26     kernel_driver = getKernelDriver(args, dev)
  27     if kernel_driver in mapping:
  28         return mapping[kernel_driver]
  29     return ""