Skip to content

Add ucxx::MemoryHandle and ucxx::RemoteKey C++ classes#190

Merged
rapids-bot[bot] merged 11 commits intorapidsai:branch-0.37from
pentschev:ucp-mem
Feb 20, 2024
Merged

Add ucxx::MemoryHandle and ucxx::RemoteKey C++ classes#190
rapids-bot[bot] merged 11 commits intorapidsai:branch-0.37from
pentschev:ucp-mem

Conversation

@pentschev
Copy link
Member

Add ucxx::MemoryHandle and ucxx::RemoteKey C++ classes, wrapping UCP memory handles (ucp_mem_h) and remote keys (ucp_rkey_h), respectively.

These new classes are required to provide basic support for RMA (Remote Memory Access) operations being implemented in #166 .

@pentschev pentschev requested review from a team as code owners February 9, 2024 20:04
@pentschev pentschev added feature request New feature or request non-breaking Introduces a non-breaking change labels Feb 9, 2024
Copy link
Contributor

@wence- wence- left a comment

Choose a reason for hiding this comment

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

Mostly some requests for clarifications in docstrings. Overall I think this looks good.

Copy link
Member Author

@pentschev pentschev left a comment

Choose a reason for hiding this comment

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

Thanks for the review @wence- , I have replied to your comments. Please have a look when you get another chance.

Copy link
Contributor

@wence- wence- left a comment

Choose a reason for hiding this comment

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

Just one outstanding question about the required size constraints on a user-provided buffer, but again, I think (if it needs it at all) this is just a doc clarification.

Thanks!

@pentschev
Copy link
Member Author

Thanks @wence- for the thorough review!

@pentschev
Copy link
Member Author

/merge

@rapids-bot rapids-bot bot merged commit d07aa81 into rapidsai:branch-0.37 Feb 20, 2024
@pentschev pentschev deleted the ucp-mem branch February 20, 2024 15:48
rapids-bot bot pushed a commit that referenced this pull request Feb 23, 2024
This PR adds support for RMA over UCP protocols, adding interfaces for the following functions: `ucp_get_nbx`, `ucp_put_nbx`, `ucp_ep_flush_nbx` and `ucp_worker_flush_nbx`. The implementation follows the same pattern use for all `ucxx::Request` operations.

Blocked waiting for #190 .

Authors:
  - Michael Demoret (https://github.com/mdemoret-nv)
  - Peter Andreas Entschev (https://github.com/pentschev)

Approvers:
  - Robert Maynard (https://github.com/robertmaynard)
  - Peter Andreas Entschev (https://github.com/pentschev)

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

Labels

feature request New feature or request non-breaking Introduces a non-breaking change

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants