AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refactor ESMI Initialization and Argument Parsing

Browse Source 
Signed-off-by: Maisam Arif <maisarif@amd.com>
Change-Id: Iefab3a8110e0d3c525ee0cef1bdef9101550e9de
This commit is contained in:
Maisam Arif
2024-02-21 03:48:09 -06:00
committed by Maisam Arif
parent 180f893791
commit f58613561c
10 changed files with 1615 additions and 1295 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/amd_smi/amd_smi_system.cc
+2 -2
View File
@@ -261,7 +261,7 @@ amdsmi_status_t AMDSmiSystem::cleanup() {
processors_.clear();
sockets_.clear();
esmi_exit();
init_flag_ = AMDSMI_INIT_ALL_PROCESSORS;
init_flag_ &= ~AMDSMI_INIT_AMD_CPUS;
}
#endif
if (init_flag_ & AMDSMI_INIT_AMD_GPUS) {
@@ -270,7 +270,7 @@ amdsmi_status_t AMDSmiSystem::cleanup() {
}
processors_.clear();
sockets_.clear();
init_flag_ = AMDSMI_INIT_ALL_PROCESSORS;
init_flag_ &= ~AMDSMI_INIT_AMD_GPUS;
rsmi_status_t ret = rsmi_shut_down();
if (ret != RSMI_STATUS_SUCCESS) {
return amd::smi::rsmi_to_amdsmi_status(ret);