Merge pull request #1001 from Sergei-Lebedev/topic/fix_sharp_team_cre…

…ate_oob

TL/SHARP: fix service team oob

src/components/tl/sharp/tl_sharp_team.c

@@ -1,5 +1,5 @@
 /**
- * Copyright (c) 2021-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+ * Copyright (c) 2021-2024, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
  *
  * See file LICENSE for terms.
  */
@@ -8,6 +8,7 @@
 #include "components/mc/ucc_mc.h"
 #include "core/ucc_ee.h"
 #include "coll_score/ucc_coll_score.h"
+#include "core/ucc_team.h"
 #include <sharp/api/version.h>
 
 UCC_CLASS_INIT_FUNC(ucc_tl_sharp_team_t, ucc_base_context_t *tl_context,
@@ -36,14 +37,23 @@ UCC_CLASS_INIT_FUNC(ucc_tl_sharp_team_t, ucc_base_context_t *tl_context,
  set.map = UCC_TL_TEAM_MAP(self);
 
  if (UCC_TL_SHARP_TEAM_LIB(self)->cfg.use_internal_oob) {
- self->oob_ctx.subset = set;
+ status = ucc_ep_map_create_nested(&UCC_TL_CORE_TEAM(self)->ctx_map,
+ &UCC_TL_TEAM_MAP(self),
+ &self->oob_ctx.subset.map);
+ if (status != UCC_OK) {
+ return status;
+ }
+ self->oob_ctx.subset.myrank = UCC_TL_TEAM_RANK(self);
  } else {
  self->oob_ctx.oob = &UCC_TL_TEAM_OOB(self);
  }
 
  status = ucc_topo_init(set, ctx->super.super.ucc_context->topo, &self->topo);
  if (UCC_OK != status) {
  tl_error(ctx->super.super.lib, "failed to init team topo");
+ if (UCC_TL_SHARP_TEAM_LIB(self)->cfg.use_internal_oob) {
+ ucc_ep_map_destroy_nested(&self->oob_ctx.subset.map);
+ }
  return status;
  }
 
@@ -156,6 +166,9 @@ UCC_CLASS_INIT_FUNC(ucc_tl_sharp_team_t, ucc_base_context_t *tl_context,
  }
  }
  ucc_topo_cleanup(self->topo);
+ if (UCC_TL_SHARP_TEAM_LIB(self)->cfg.use_internal_oob) {
+ ucc_ep_map_destroy_nested(&self->oob_ctx.subset.map);
+ }
  return status;
 }
 
@@ -180,6 +193,9 @@ UCC_CLASS_CLEANUP_FUNC(ucc_tl_sharp_team_t)
  }
  }
  }
+ if (UCC_TL_SHARP_TEAM_LIB(self)->cfg.use_internal_oob) {
+ ucc_ep_map_destroy_nested(&self->oob_ctx.subset.map);
+ }
 }
 
 UCC_CLASS_DEFINE_DELETE_FUNC(ucc_tl_sharp_team_t, ucc_base_team_t);