Skip to content

Conversation

@tvegas1
Copy link
Contributor

@tvegas1 tvegas1 commented Jun 7, 2024

replaces first v5.0.x backport #12579

cherry-pick from main #12577 and #12590

What

The UCX PML needs to differentiate between mpi serialized and single thread mode according to MPI standard.

Why

Using UCX with multiple threads, even when providing mutual exclusion, is not completely equivalent to only using UCX with one same thread.

To guarantee ordering and avoid corruption, some UCX transports need the cpu to flush stores to the devices, before another thread can progress the worker. For latency reasons, this needed flush is only performed in non single thread mode.

@github-actions github-actions bot added this to the v5.0.4 milestone Jun 7, 2024
@github-actions
Copy link

github-actions bot commented Jun 7, 2024

Hello! The Git Commit Checker CI bot found a few problems with this PR:

637b40e: UCX: Propagate MPI serialized for all worker creat...

  • check_cherry_pick: does not include a cherry pick message (did you need to bot:notacherrypick?)

df838c7: pml/ucx: Propagate MPI serialized thread mode

  • check_cherry_pick: does not include a cherry pick message (did you need to bot:notacherrypick?)

Please fix these problems and, if necessary, force-push new commits back up to the PR branch. Thanks!

tvegas1 added 2 commits June 7, 2024 20:03
The UCX PML needs to differentiate between mpi serialized and single thread
mode according to MPI standard.

Using UCX with multiple threads, even when providing mutual exclusion, is not
completely equivalent to only using UCX with one same thread.

Signed-off-by: Thomas Vegas <[email protected]>
(cherry picked from commit b33b61a)
Move MPI to UCX thread mode function to common source. Also use
serialized mode for all oshmem initializations.

Signed-off-by: Thomas Vegas <[email protected]>
(cherry picked from commit efa293b)
@tvegas1 tvegas1 force-pushed the ucx_thread_mode_combined_v5.0.x branch from 637b40e to 44cc663 Compare June 7, 2024 17:04
@tvegas1
Copy link
Contributor Author

tvegas1 commented Jun 10, 2024

@brminich, @bosilca, @yosefe, can you please review?

Copy link
Member

@bosilca bosilca left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants