Merge pull request #889 from Sergei-Lebedev/topic/fix_alltoall_hang

TEST: progress ucc context during skip reduce
openucx · Dec 13, 2023 · e20dbb5 · e20dbb5
2 parents d257388 + a00f966
commit e20dbb5
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/test/mpi/test_case.cc b/test/mpi/test_case.cc
@@ -149,7 +149,14 @@ test_skip_cause_t TestCase::skip_reduce(int skip_cond, test_skip_cause_t cause,
 {
     test_skip_cause_t test_skip;
     test_skip_cause_t skip = skip_cond ? cause : TestCase::test_skip;
-    MPI_Allreduce((void*)&skip, (void*)&test_skip, 1, MPI_INT, MPI_MAX, comm);
+    MPI_Request req;
+    int completed;
+
+    MPI_Iallreduce((void*)&skip, (void*)&test_skip, 1, MPI_INT, MPI_MAX, comm, &req);
+    do {
+        MPI_Test(&req, &completed, MPI_STATUS_IGNORE);
+        tc_progress_ctx();
+    } while(!completed);
     TestCase::test_skip = test_skip;
     return test_skip;
 }