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

[Distributed] [16/N] Fix clang-tidy warnings in torch/csrc/distributed/c10d (#137404)

Browse Source 
Follows #137072

Pull Request resolved: https://github.com/pytorch/pytorch/pull/137404
Approved by: https://github.com/Skylion007
This commit is contained in:
cyy
2024-10-10 18:05:32 +00:00
committed by PyTorch MergeBot
parent 20815c7cb9
commit 94e12f97dc
22 changed files with 256 additions and 289 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -636,9 +636,9 @@ struct NCCLTraceBuffer {
std::optional<c10::time_t> time_discovered_completed_;
// size information for input/output tensors
c10::SmallVector<int, 4> input_dims_;
c10::SmallVector<int64_t, 4> input_dims_;
std::vector<c10::ScalarType> input_dtypes_;
c10::SmallVector<int, 4> output_dims_;
c10::SmallVector<int64_t, 4> output_dims_;
std::vector<c10::ScalarType> output_dtypes_;
c10::SmallVector<int64_t, 8> sizes_; // flattened from inputs, outputs
bool retired_ = false; // is this work entry no longer in the workMetaList_?