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

Apply clang-format to distributed/c10d folder (#107140)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/107140
Approved by: https://github.com/H-Huang
This commit is contained in:
Shen Li
2023-08-14 06:59:47 -07:00
committed by PyTorch MergeBot
parent 858b465d74
commit dd6319198d
28 changed files with 370 additions and 307 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
torch/csrc/distributed/c10d/ParamCommsUtils.hpp
View File

@ -1,29 +1,27 @@
#pragma once
#include <string>
#include <vector>
#include <ATen/core/ivalue.h>
#include <ATen/record_function.h>
#include <c10/macros/Macros.h>
#include <c10/util/ThreadLocalDebugInfo.h>
#include <ATen/record_function.h>
#include <ATen/core/ivalue.h>
#include <string>
#include <vector>
namespace torch {
extern TORCH_API const std::string kParamCommsCallName;
class TORCH_API ParamCommsDebugInfo
: public c10::DebugInfoBase {
class TORCH_API ParamCommsDebugInfo : public c10::DebugInfoBase {
public:
ParamCommsDebugInfo() = default;
ParamCommsDebugInfo(
int rank,
std::string&& colName,
int inSize,
int outSize,
at::ScalarType dType,
std::vector<int64_t> inSplitSizes,
std::vector<int64_t> outSplitSizes);
int rank,
std::string&& colName,
int inSize,
int outSize,
at::ScalarType dType,
std::vector<int64_t> inSplitSizes,
std::vector<int64_t> outSplitSizes);
~ParamCommsDebugInfo() override = default;
@ -80,7 +78,7 @@ class TORCH_API ParamCommsDebugInfo
c10::DebugInfoGuard g(c10::DebugInfoKind::PARAM_COMMS_INFO, paramCommsInfo); \
std::initializer_list<const c10::IValue> paramList = { \
c10::IValue(seq), \
c10::IValue(pg_ptr), \
c10::IValue(pg_ptr), \
rank, \
colName, \
inSplitSizes, \
@ -91,8 +89,8 @@ class TORCH_API ParamCommsDebugInfo
#define RECORD_PARAM_COMMS_DATA( \
seq, \
pg_ptr, \
InputTensors, \
OutputTensors, \
InputTensors, \
OutputTensors, \
rank, \
colName, \
inSize, \
@ -104,7 +102,7 @@ class TORCH_API ParamCommsDebugInfo
rank, colName, inSize, outSize, dType, inSplitSizes, outSplitSizes); \
c10::DebugInfoGuard g(c10::DebugInfoKind::PARAM_COMMS_INFO, paramCommsInfo); \
std::initializer_list<const c10::IValue> paramList = { \
c10::IValue(InputTensors), \
c10::IValue(InputTensors), \
c10::IValue(seq), \
c10::IValue(pg_ptr), \
rank, \