pytorch/caffe2
Ke Wen 8fcda2c60d [SymmMem] Add runtime detection of NVSHMEM (#156291)
so that we can pick the default backend for SymmetricMemory without
fully relying on env var `TORCH_SYMMMEM=CUDA | NVSHMEM`

On Python side, the following API is added:
`torch.distributed._symmetric_memory.is_nvshmem_available()`

Pull Request resolved: https://github.com/pytorch/pytorch/pull/156291
Approved by: https://github.com/Skylion007
ghstack dependencies: #155506, #155835, #155968, #155971, #155975, #156116, #156117
2025-06-19 08:26:11 +00:00
..
core Record the XPU and XCCL build settings in the compiled binary (#147161) 2025-05-20 09:21:39 +00:00
perfkernels Optimize SVE embedding performance (#150176) 2025-04-07 18:01:54 +00:00
serialize Update PyTorchStreamReader API to take cpu allocator override (#150439) 2025-04-18 01:53:14 +00:00
utils [3/N] Use internal linkage in C++ files (#151297) 2025-05-05 17:48:39 +00:00
.clang-format [BE][clang-format] make macro PyObject_HEAD_INIT(type) and PyVarObject_HEAD_INIT(type, size) have its own line (#136949) 2024-10-02 18:39:22 +00:00
CMakeLists.txt [SymmMem] Add runtime detection of NVSHMEM (#156291) 2025-06-19 08:26:11 +00:00
unexported_symbols.lds
version_script.lds