fix getting device from empty list

This commit is contained in:
Nicolas Mowen 2024-10-07 06:49:44 -06:00
parent dcaed0e90f
commit 7029a63166

View File

@ -31,6 +31,9 @@ class LibvaGpuSelector:
devices = list(filter(lambda d: d.startswith("render"), os.listdir("/dev/dri"))) devices = list(filter(lambda d: d.startswith("render"), os.listdir("/dev/dri")))
if not devices:
return "/dev/dri/renderD128"
if len(devices) < 2: if len(devices) < 2:
self._selected_gpu = f"/dev/dri/{devices[0]}" self._selected_gpu = f"/dev/dri/{devices[0]}"
return self._selected_gpu return self._selected_gpu