Fix P2P scheduling (#560)

[ROCm/rccl commit: 5cb2aca3d9]
2022-06-06 13:32:28 -07:00
parent f2dbc77afe
commit 11a6cdd52f
4 changed files with 95 additions and 68 deletions
@@ -412,6 +412,8 @@ ncclResult_t ncclProxyComputeP2p(struct ncclInfo* info, struct ncclProxyOp* op)
  info->chunkSize = stepSize;
  op->root = info->root;
  op->nbytes = info->count;
+  if (info->root == -1) return ncclSuccess;
+
  struct ncclPeer* peer = channel->peers + op->root;

  if (info->coll == ncclFuncSend) {