Limit P2P channels on Rome

[ROCm/rccl commit: 42955f5f4f]
2020-09-17 16:01:45 -07:00
parent f6b08ca547
commit f0a303664e
5 changed files with 22 additions and 12 deletions
@@ -304,7 +304,7 @@ group_cleanup:
        *args->init.newcomm = NULL;
      } else {
        struct ncclComm* comm = args->coll.comm;
-        for (int c=0; c<comm->p2pnChannels; c++) {
+        for (int c=0; c<std::max(comm->nChannels, comm->p2pnChannels); c++) {
          struct ncclChannel* channel = comm->channels+c;
          for (int i=0; i<channel->collCount; i++) {
            channel->collectives[(channel->collStart + i)%NCCL_MAX_OPS].active = 0;