pmodels / mpich

Official MPICH Repository
http://www.mpich.org
Other
562 stars 279 forks source link

ch4: add MPIDI_NM_am_tag_send/recv #7183

Closed hzhou closed 3 weeks ago

hzhou commented 1 month ago

Pull Request Description

In mpidig RNDV, if the netmod supports tag -- query via MPIDI_NM_am_can_do_tag() -- add MPIDI_NM_amtag{send,recv} to support direct send/recv. This not only avoids the extra copy in UCX or extra hdr message and additional RDMA read in OFI am_send_long, it also supports native GPU RDMA direct when the netmod supports it.

This is how it works -

Strength

image

Comparisons

Follow-up

[skip warnings]

Author Checklist

hzhou commented 3 weeks ago

test:mpich/ch4/most ✔️ test:mpich/ch4/ofi/more - am-only has some issue

hzhou commented 3 weeks ago

test:mpich/ch4/ofi/more ✔️