Use rocm-smi thread only mutex when available (#1169)
[ROCm/rccl commit: a64aab5f63]
This commit is contained in:
@@ -35,7 +35,11 @@ THE SOFTWARE.
|
||||
|
||||
|
||||
ncclResult_t rocm_smi_init() {
|
||||
#ifdef USE_ROCM_SMI_THREAD_ONLY_MUTEX
|
||||
ROCMSMICHECK(rsmi_init(RSMI_INIT_FLAG_THRAD_ONLY_MUTEX));
|
||||
#else
|
||||
ROCMSMICHECK(rsmi_init(0));
|
||||
#endif
|
||||
rsmi_version_t version;
|
||||
ROCMSMICHECK(rsmi_version_get(&version));
|
||||
INFO(NCCL_INIT, "rocm_smi_lib: version %d.%d.%d.%s", version.major, version.minor, version.patch, version.build);
|
||||
|
||||
Reference in New Issue
Block a user