Udpated amdsmi_quick_start.py with cpus preloaded

Signed-off-by: Maisam Arif <Maisam.Arif@amd.com>
Change-Id: I4a5ca0d30d2fce3b4fa3a6a13599a18b0dd16ce7
Этот коммит содержится в:
Maisam Arif
2024-09-05 09:45:02 -05:00
родитель 397d8d9339
Коммит 8b3d45e301
+4 -2
Просмотреть файл
@@ -45,6 +45,8 @@ signal.signal(signal.SIGINT, signal_handler)
signal.signal(signal.SIGTERM, signal_handler)
atexit.register(amdsmi_shut_down)
devices = amdsmi_get_processor_handles()
gpus = amdsmi_get_socket_handles()
cpus = amdsmi_get_cpusocket_handles()
print(f"devices variable populated with:{devices}")
print(f"gpus variable populated with:{gpus}")
print(f"cpus variable populated with:{cpus}")