Symmetric memory pytorch backends#6023
Merged
Merged
Conversation
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Adds support for PyTorch-backed symmetric memory in nvFuser alongside the existing native implementation. This enables using c10d::symmetric_memory (NCCL/NVSHMEM/CUDA) as an alternative backend while keeping the current SymmetricTensor API unchanged.
--Default behavior unchanged (native backend).
--PyTorch backend is opt-in via NVFUSER_ENABLE=symmetric_memory_backend(...).
allocate (empty_strided_p2p)
setupRemoteHandles (c10d::symmetric_memory::rendezvous)
remoteTensor(get_remote_tensor)