Added all option to gpu selection

Change-Id: Ifcb49d66239c54fd3a214f4782321bbcc59e057a
Signed-off-by: Maisam Arif <maisarif@amd.com>
Этот коммит содержится в:
Maisam Arif
2023-09-24 02:54:06 -05:00
коммит произвёл Maisam Arif
родитель 5ad17006b2
Коммит a28df4a242
+8 -1
Просмотреть файл
@@ -151,7 +151,11 @@ class AMDSMIHelpers():
"UUID": uuid,
"Device Handle": device_handle,
}
gpu_choices_str += f"ID:{gpu_id:<2} | BDF:{bdf} | UUID:{uuid}\n"
gpu_choices_str += f"ID:{gpu_id} | BDF:{bdf} | UUID:{uuid}\n"
# Add the all option to the gpu_choices
gpu_choices["all"] = "all"
gpu_choices_str += " all | Selects all devices\n"
return (gpu_choices, gpu_choices_str)
@@ -168,6 +172,9 @@ class AMDSMIHelpers():
amdsmi device_handles
(False, str): Return False, and the first input that failed to be converted
"""
if gpu_selections == ["all"]:
return (True, amdsmi_interface.amdsmi_get_processor_handles())
if isinstance(gpu_selections, str):
gpu_selections = [gpu_selections]