NVIDIA-NeMo / NVIDIA-NeMo/RL

Symmetric memory enablement for non-colocated refit

Open
#1,399 0 comments 0 reactions 1 assignee Claimed by @youngeunkwon0405 View on GitHub
enhancement Performance
Dominant language
Python
Stars
2k
Forks
561
Avg merge
4d 5h
Merged PRs (30d)
145

Description

**Is your feature request related to a problem? Please describe.**

vLLM supports symmetric memory enablement API from [v0.11.0](https://docs.vllm.ai/en/latest/api/vllm/distributed/device_communicators/pynccl_allocator.html#vllm.distributed.device_communicators.pynccl_allocator.nccl_symm_mem_context).
With NCCL v2.29 (next release), symmetric memory enablement will become beneficial IB domain communication as well.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.