Skip to content

Commit d83caaa

Browse files
authored
TOOLS/UCC_PERFTEST: Adding missing progress calls (#1151)
Add ucc_context_progress() calls to team creation and destruction loops to prevent hangs when collective operations need progress to complete. This fixes a hang in team creatation observed during the mcast-enabled ucc_perftest.
1 parent b593e66 commit d83caaa

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

tools/perf/ucc_pt_comm.cc

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -169,6 +169,9 @@ ucc_status_t ucc_pt_comm::init()
169169
free_ctx, st);
170170
do {
171171
st = ucc_team_create_test(team);
172+
if (st == UCC_INPROGRESS) {
173+
ucc_context_progress(context);
174+
}
172175
} while(st == UCC_INPROGRESS);
173176
UCCCHECK_GOTO(st, free_ctx, st);
174177
ucc_context_config_release(ctx_config);
@@ -215,6 +218,9 @@ ucc_status_t ucc_pt_comm::finalize()
215218

216219
do {
217220
status = ucc_team_destroy(team);
221+
if (status == UCC_INPROGRESS) {
222+
ucc_context_progress(context);
223+
}
218224
} while (status == UCC_INPROGRESS);
219225
if (status != UCC_OK) {
220226
std::cerr << "ucc team destroy error: " << ucc_status_string(status);

0 commit comments

Comments
 (0)