Searched refs:nccl_done (Results 1 – 1 of 1) sorted by relevance
122 Notification nccl_done; in Run() local140 auto done_callback = [&nccl_done, &nccl_status](const Status& s) { in Run()142 nccl_done.Notify(); in Run()191 nccl_done.WaitForNotification(); in Run()