Moved mscclpp_ncclGetUniqueId call into ncclCommInitRankFunc (#1332)

* Moved call to `mscclpp_ncclGetUniqueId` into `ncclCommInitRankFunc` to avoid setting up transport early in environments where MSCCL++ isn't valid.

* Checking `mscclEnabled` for the process and the topology to gate MSCCL++.

* Allowed `mscclForceEnable` to enable MSCCL++.

[ROCm/rccl commit: 853a0586b4]
Этот коммит содержится в:
corey-derochie-amd
2024-09-16 16:41:40 -06:00
коммит произвёл GitHub
родитель 31b576b969
Коммит cf48e57bd9
3 изменённых файлов: 38 добавлений и 33 удалений
+2
Просмотреть файл
@@ -49,4 +49,6 @@ namespace std {
bool operator ==(const ncclUniqueId& a, const ncclUniqueId& b);
bool mscclppCommCompatible(ncclComm_t comm);
#endif