Update src/components/tl/ucp/tl_ucp_team.c

Co-authored-by: samnordmann <snordmann@nvidia.com>
Sergei-Lebedev · Mar 6, 2024 · 2f3db2e · 2f3db2e
1 parent a8ae718
commit 2f3db2e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/components/tl/ucp/tl_ucp_team.c b/src/components/tl/ucp/tl_ucp_team.c
@@ -91,7 +91,7 @@ UCC_CLASS_INIT_FUNC(ucc_tl_ucp_team_t, ucc_base_context_t *tl_context,
     }
 
     if (self->topo && !IS_SERVICE_TEAM(self) && self->topo->topo->sock_bound) {
-        max_radix = ucc_min(UCC_TL_TEAM_SIZE(self),
+        max_radix = ucc_topo_max_ppn(self->topo) == 1 ? UCC_TL_TEAM_SIZE(self) : ucc_min(UCC_TL_TEAM_SIZE(self),ucc_topo_min_socket_size(self->topo));
                             ucc_topo_max_ppn(self->topo) == 1 ?
                             UCC_TL_TEAM_SIZE(self):
                             ucc_topo_min_socket_size(self->topo));