From 52c73a95f78dfd73e8dfc7e08d96c54c1a3a456c Mon Sep 17 00:00:00 2001 From: Sergey Lebedev Date: Fri, 3 May 2024 09:24:17 +0000 Subject: [PATCH] TL/NCCL: disable active set colls for lazyinit --- src/components/tl/nccl/tl_nccl_coll.c | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/components/tl/nccl/tl_nccl_coll.c b/src/components/tl/nccl/tl_nccl_coll.c index ee3d523b0b..ab05515e86 100644 --- a/src/components/tl/nccl/tl_nccl_coll.c +++ b/src/components/tl/nccl/tl_nccl_coll.c @@ -145,6 +145,10 @@ ucc_status_t ucc_tl_nccl_init_task(ucc_base_coll_args_t *coll_args, } if (ucc_unlikely(nccl_team->comm_state != TL_NCCL_COMM_STATE_READY)) { + if (UCC_COLL_ARGS_ACTIVE_SET(&coll_args->args)) { + /* active set is not supported with lazy comm init*/ + return UCC_ERR_NOT_SUPPORTED; + } status = ucc_tl_nccl_comm_init(nccl_team); if (ucc_unlikely(status != UCC_OK)) { return status;