Skip to content

Add support for ucc_mem_map and ucc_mem_unmap #1070

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged

Conversation

wfaderhold21
Copy link
Collaborator

What

Implementation of PR #1037

NOTE: branch is branched from #1037 and will need to be rebased before merging

@swx-jenkins3
Copy link

Can one of the admins verify this patch?

@wfaderhold21 wfaderhold21 force-pushed the topic/explicit-mem-map-impl branch from 61ab0da to 4a2f9c6 Compare January 16, 2025 00:31
@manjugv
Copy link
Contributor

manjugv commented Jan 18, 2025

@janjust @nsarka Can you test on your setup? Ask @wfaderhold21 for details.

@wfaderhold21
Copy link
Collaborator Author

Example user usage of API changes: https://github.com/wfaderhold21/simple-ucc-ex

@wfaderhold21 wfaderhold21 force-pushed the topic/explicit-mem-map-impl branch from 23d5050 to 9691c1f Compare March 4, 2025 21:52
@wfaderhold21
Copy link
Collaborator Author

@Sergei-Lebedev Thank you for the feedback. I believe I have addressed all your concerns. Please let me know if I haven't.

@wfaderhold21
Copy link
Collaborator Author

@janjust do we need more code review concerning this PR or can we apply the completed label?

@Sergei-Lebedev
Copy link
Contributor

@Sergei-Lebedev Thank you for the feedback. I believe I have addressed all your concerns. Please let me know if I haven't.

thanks, I reviewed only part of this PR since it's big. I will continue reviewing this week.

ferrol aderholdt added 26 commits June 12, 2025 10:49
CL/BASIC: add support for ucc_mem_map/unmap

CL/HIER: add support for ucc_mem_map/unmap

CL/DOCA_UROM: add support for ucc_mem_map/unmap

TL/CUDA: add support for ucc_mem_map/unmap

TL/MLX5: add support for ucc_mem_map/unmap

TL/NCCL: add support for ucc_mem_map/unmap

TL/RCCL: add support for ucc_mem_map/unmap

TL/SELF: add support for ucc_mem_map/unmap

TL/SHARP: add support for ucc_mem_map/unmap
TL/UCP: add support for communication with memmap

REVIEW: various fixes

TL/UCP: Enable multiple packed buffers
TL/UCP: additional support for memh in onesided alltoall
@janjust janjust force-pushed the topic/explicit-mem-map-impl branch from 4b7b11d to 0870c18 Compare June 12, 2025 15:49
@Sergei-Lebedev
Copy link
Contributor

ok to test

@Sergei-Lebedev Sergei-Lebedev merged commit 9f72275 into openucx:master Jun 16, 2025
9 checks passed
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.

7 participants