mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
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 |
||
|---|---|---|
| .. | ||
| core | ||
| perfkernels | ||
| serialize | ||
| utils | ||
| .clang-format | ||
| CMakeLists.txt | ||
| unexported_symbols.lds | ||
| version_script.lds | ||