From 1893cc36fee03aa607f235299954faebd54c969d Mon Sep 17 00:00:00 2001 From: Ferrol Aderholdt Date: Wed, 15 Nov 2023 03:32:15 -0800 Subject: [PATCH] TL/UCP: poll for onesided alltoall completion (#876) Co-authored-by: ferrol aderholdt --- 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; }