From b3fc81d2b8f128d722bcefe104387ae325287595 Mon Sep 17 00:00:00 2001 From: ferrol aderholdt Date: Wed, 8 Nov 2023 08:59:46 -0800 Subject: [PATCH] TL/UCP: poll for onesided alltoall completion --- src/components/tl/ucp/alltoall/alltoall_onesided.c | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/src/components/tl/ucp/alltoall/alltoall_onesided.c b/src/components/tl/ucp/alltoall/alltoall_onesided.c index 99c56d281c..856b392534 100644 --- a/src/components/tl/ucp/alltoall/alltoall_onesided.c +++ b/src/components/tl/ucp/alltoall/alltoall_onesided.c @@ -55,9 +55,7 @@ void ucc_tl_ucp_alltoall_onesided_progress(ucc_coll_task_t *ctask) ucc_rank_t gsize = UCC_TL_TEAM_SIZE(team); long * pSync = TASK_ARGS(task).global_work_buffer; - if ((*pSync < gsize) || - (task->onesided.put_completed < task->onesided.put_posted)) { - ucp_worker_progress(UCC_TL_UCP_TEAM_CTX(team)->worker.ucp_worker); + if (ucc_tl_ucp_test_onesided(task, gsize) == UCC_INPROGRESS) { return; }