Merge pull request #9291 from awlauria/waitall_master

awlauria · web-flow · commit a99747f80f5b · 2021-08-19T08:43:52.000-04:00
Improve MPI_Waitall performance for MPI_THREAD_MULTIPLE
diff --git a/ompi/request/req_wait.c b/ompi/request/req_wait.c
@@ -243,7 +243,7 @@ int ompi_request_default_wait_all( size_t count,
             continue;
         }
 
-        if (!OPAL_ATOMIC_COMPARE_EXCHANGE_STRONG_PTR(&request->req_complete, &_tmp_ptr, &sync)) {
+        if (REQUEST_COMPLETE(request) || !OPAL_ATOMIC_COMPARE_EXCHANGE_STRONG_PTR(&request->req_complete, &_tmp_ptr, &sync)) {
             if( OPAL_LIKELY( REQUEST_COMPLETE(request) ) ) {
                 if( OPAL_UNLIKELY( MPI_SUCCESS != request->req_status.MPI_ERROR ) ) {
                     failed++;

Original file line number	Diff line number	Diff line change
`@@ -243,7 +243,7 @@ int ompi_request_default_wait_all( size_t count,`
`243`	`243`	`continue;`
`244`	`244`	`}`
`245`	`245`
`246`		`- if (!OPAL_ATOMIC_COMPARE_EXCHANGE_STRONG_PTR(&request->req_complete, &_tmp_ptr, &sync)) {`
	`246`	`+ if (REQUEST_COMPLETE(request) \|\| !OPAL_ATOMIC_COMPARE_EXCHANGE_STRONG_PTR(&request->req_complete, &_tmp_ptr, &sync)) {`
`247`	`247`	`if( OPAL_LIKELY( REQUEST_COMPLETE(request) ) ) {`
`248`	`248`	`if( OPAL_UNLIKELY( MPI_SUCCESS != request->req_status.MPI_ERROR ) ) {`
`249`	`249`	`failed++;`