tools/helpers/gpu.py

   1 import glob
   2 import os
   3 import tools.helpers.props
   4
   5 unsupported = ["nvidia"]
   6
   7 def getMinor(args, dev):
   8     return tools.helpers.props.file_get(args, "/sys/class/drm/{}/uevent".format(dev), "MINOR")
   9
  10 def getKernelDriver(args, dev):
  11     return tools.helpers.props.file_get(args, "/sys/class/drm/{}/device/uevent".format(dev), "DRIVER")
  12
  13 def getCardFromRender(args, dev):
  14     try:
  15         return "/dev/dri/" + os.path.basename(glob.glob("/sys/class/drm/{}/device/drm/card*".format(dev))[0])
  16     except IndexError:
  17         return ""
  18
  19 def getDriNode(args):
  20     for node in glob.glob("/dev/dri/renderD*"):
  21         renderDev = os.path.basename(node)
  22         if getKernelDriver(args, renderDev) not in unsupported:
  23             return node, getCardFromRender(args, renderDev)
  24     return "", ""
  25
  26 def getVulkanDriver(args, dev):
  27     mapping = {
  28         "i915": "intel",
  29         "amdgpu": "radeon",
  30         "radeon": "radeon",
  31         "panfrost": "panfrost",
  32         "msm": "freedreno",
  33         "msm_dpu": "freedreno",
  34         "vc4": "broadcom",
  35         "nouveau": "nouveau",
  36     }
  37     kernel_driver = getKernelDriver(args, dev)
  38
  39     if kernel_driver == "i915":
  40         try:
  41             gen = tools.helpers.run.user(args,["awk", "/^graphics version:|^gen:/ {print $NF}",
  42                 "/sys/kernel/debug/dri/{}/i915_capabilities".format(getMinor(args, dev))], output_return=True, check=False)
  43             if int(gen) < 9:
  44                 return "intel_hasvk"
  45         except:
  46             pass
  47
  48     if kernel_driver in mapping:
  49         return mapping[kernel_driver]
  50     return ""