TL/NCCL: disable for team size 1 (openucx#706)

edgargabriel · Jan 10, 2023 · e9eb0db · e9eb0db
1 parent 0eb1088
commit e9eb0db
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/components/tl/nccl/tl_nccl_team.c b/src/components/tl/nccl/tl_nccl_team.c
@@ -22,6 +22,12 @@ UCC_CLASS_INIT_FUNC(ucc_tl_nccl_team_t, ucc_base_context_t *tl_context,
     UCC_CLASS_CALL_SUPER_INIT(ucc_tl_team_t, &ctx->super, params);
 
     size = UCC_TL_TEAM_SIZE(self);
+    if (size < 2) {
+        tl_debug(tl_context->lib,
+                 "team size %d is too small, minimal size is 2",
+                 UCC_TL_TEAM_SIZE(self));
+        return UCC_ERR_NOT_SUPPORTED;
+    }
     self->comm_state = UCC_OK;
     self->unique_id  = ucc_malloc(sizeof(ncclUniqueId) * (size + 1),
                                   "tl_nccl_unique_id");