frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Don't include cuh header when USE_NVSHMEM is off (#162635)

Browse Source 
Signed-off-by: Edward Yang <ezyang@meta.com>

Pull Request resolved: https://github.com/pytorch/pytorch/pull/162635
Approved by: https://github.com/kwen2501
This commit is contained in:
Edward Yang
2025-09-10 16:28:59 -04:00
committed by PyTorch MergeBot
parent 4fd2a2b273
commit da5069f289
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
torch/csrc/distributed/c10d/init.cpp
View File

@ -48,7 +48,10 @@
#include <torch/csrc/distributed/c10d/PrefixStore.hpp>
#include <torch/csrc/distributed/c10d/symm_mem/DMAConnectivity.hpp>
#include <torch/csrc/distributed/c10d/symm_mem/SymmetricMemory.hpp>
#ifdef USE_NVSHMEM
#include <torch/csrc/distributed/c10d/symm_mem/nvshmem_extension.cuh>
#endif
#include <torch/csrc/distributed/c10d/comm.hpp>
#include <torch/csrc/distributed/c10d/debug.h>