Init the amdsmi using rocm_smi for libdrm

Init the ams_smi using the rocm-smi, which makes the GPU discovery
consistent with or without libdrm.

Change-Id: Ic714781f8ce791451b0c057621525926edb7f5ee
This commit is contained in:
Bill(Shuzhou) Liu
2022-11-07 11:05:20 -06:00
parent c8caa80405
commit b34b7451e8
2 changed files with 71 additions and 85 deletions
+3 -1
View File
@@ -76,7 +76,9 @@ class AMDSmiSystem {
amdsmi_device_handle* device_handle);
private:
AMDSmiSystem() : init_flag_(AMDSMI_INIT_ALL_DEVICES) {}
AMDSmiSystem() : init_flag_(AMDSMI_INIT_AMD_GPUS) {}
amdsmi_status_t get_gpu_bdf_by_index(uint32_t index, std::string& bdf);
amdsmi_status_t populate_amd_gpu_devices();
uint64_t init_flag_;
AMDSmiDrm drm_;
std::vector<AMDSmiSocket*> sockets_;