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

Enable querying the build and runtime NCCL versions (#156305)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/156305
Approved by: https://github.com/wconstab, https://github.com/Skylion007, https://github.com/fegin
This commit is contained in:
Luca Wehrstedt
2025-06-18 12:48:24 +00:00
committed by PyTorch MergeBot
parent bc9bd2a766
commit d4ad280429
4 changed files with 35 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
torch/csrc/distributed/c10d/NCCLUtils.hpp
View File

@ -229,6 +229,7 @@ static std::map<at::ScalarType, ncclDataType_t> ncclDataType = {
TORCH_API size_t hashTensors(const std::vector<at::Tensor>& tensors);
TORCH_API std::string getNcclVersion();
TORCH_API std::tuple<int, int, int> getNcclVersionTuple();
TORCH_API int getNcclVersionNumber();
TORCH_API std::string ncclGetErrorWithVersion(ncclResult_t error);
int nccl_nonblocking_timeout();