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.

Sign up for GitHub

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

Jump to bottom

Implement CUDA/VMM memory detection in UCX backend #75

Open

artpol84 opened this issue Mar 20, 2025 · 0 comments

Labels

Contributor

artpol84 commented Mar 20, 2025

Currently, we only detect cuda-malloc'd memory:
https://github.com/ai-dynamo/nixl/blob/main/src/plugins/ucx/ucx_backend.cpp#L50

UCX is also capable of detecting VMM memory:
https://github.com/openucx/ucx/blob/master/src/uct/cuda/cuda_copy/cuda_copy_md.c#L633

This code needs to be proted to NIXL too

saturley-hall added the enhancement label

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