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


[ROCm/amdsmi commit: b34b7451e8]
This commit is contained in:
Bill(Shuzhou) Liu committed 2022-11-07 11:09:09 -06:00
1 parent 17fa3924d6
commit 2d0fb9744a
2 files changed
+71 -85

No files matched your search

@@ -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_;