UCP: Add API to query datatype attributes#8120
Conversation
|
Hi @rdietric. Thanks for sharing your API proposal. Do you have signed CLA with the project ? Corporate and private CLAs can be found here: https://openucx.org/license/ Guidance to contributors is here: https://github.com/openucx/ucx/wiki/Guidance-for-contributors |
|
I used my existing Github account and just added NVIDIA to my profile. Sorry for the inconveniences! |
@rdietric not a problem, you did better than many others ;) |
aef2c05 to
33ea8ac
Compare
|
@tonycurtis can you please review documentation. |
|
Looks good |
|
I added PR #8150 with the implementation. |
|
Is there anything else I can do to get the PR merged? |
|
@shamisp @tonycurtis can you pls review as well? |
|
@rdietric pls check the build failures |
|
@tonycurtis can you pls take a look? |
|
@rdietric pls squash |
0386df2
e868ac6 to
0386df2
Compare
|
I squashed the commits. The failing tests seem unrelated to changes in this PR. |
What
This PR adds a new public API function to query attributes for a UCP datatype. Currently, the packed size is the only attribute that can be queried.
Why ?
The size of a message or data transfer is a performance-relevant information. Performance tools should be able to query this information. MPI provides
MPI_Type_sizeto query the size of any MPI datatype.How ?
A follow-up PR will implement
ucp_dt_queryinsrc/ucp/dt/dt.c. Unfortunately, the size of a UCP generic datatype is not available before it has been used in a UCP communication operation without more significant API changes.