Этот коммит содержится в:
Ziyue Yang
2023-05-24 22:41:05 +08:00
коммит произвёл GitHub
родитель ed252c30f4
Коммит 7d6e7bcd7d
10 изменённых файлов: 167 добавлений и 122 удалений
+1
Просмотреть файл
@@ -622,6 +622,7 @@ static ncclResult_t devCommSetup(ncclComm_t comm) {
// Init NPKit
NCCLCHECK(NpKit::Init(comm->rank));
tmpCommAndChans.comm.npKitEventCollectContexts = NpKit::GetGpuEventCollectContexts();
tmpCommAndChans.comm.cpuTimestamp = NpKit::GetCpuTimestamp();
#endif
#ifdef ENABLE_PROFILING