Use rocm-smi thread only mutex when available (#1169)

src/misc/rocm_smi_wrap.cc

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