From ae9d7a098c86ef3c8e575cc872d4466c31c196d9 Mon Sep 17 00:00:00 2001 From: Maisam Arif Date: Thu, 5 Sep 2024 09:45:02 -0500 Subject: [PATCH] Udpated amdsmi_quick_start.py with cpus preloaded Signed-off-by: Maisam Arif Change-Id: I4a5ca0d30d2fce3b4fa3a6a13599a18b0dd16ce7 [ROCm/amdsmi commit: 8b3d45e3011e7deb797e8acd952041c685102d54] --- projects/amdsmi/tools/amdsmi_quick_start.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/projects/amdsmi/tools/amdsmi_quick_start.py b/projects/amdsmi/tools/amdsmi_quick_start.py index 2108e29f91..caa3bece26 100644 --- a/projects/amdsmi/tools/amdsmi_quick_start.py +++ b/projects/amdsmi/tools/amdsmi_quick_start.py @@ -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}")