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

Apply clang-format to RPC files (#34139)

Browse Source 
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/34139

Test Plan: Imported from OSS

Differential Revision: D20227342

Pulled By: mrshenli

fbshipit-source-id: 01b478bde1f6a51f69eb5277fa90ba6ac2d4b5dc
This commit is contained in:
Shen Li
2020-03-03 16:35:55 -08:00
committed by Facebook Github Bot
parent 3af0dffe84
commit 7da24b36b1
5 changed files with 32 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
torch/csrc/distributed/rpc/rref_impl.h
View File

@ -44,7 +44,6 @@ struct TORCH_API RRefForkData {
std::string typeStr);
};
// Note [RRef Protocol]
// ~~~~~~~~~~~~~~~~~~~~~~~~~~
//
@ -210,7 +209,7 @@ class TORCH_API RRef : public RRefInterface {
inline bool isPyObj() {
return type_ == PyObjectType::get();
}
inline const TypePtr type() const override{
inline const TypePtr type() const override {
return type_;
}
@ -240,7 +239,11 @@ class TORCH_API UserRRef final : public RRef {
UserRRef& operator=(const UserRRef& other) = delete;
UserRRef& operator=(UserRRef&& other) = delete;
UserRRef(worker_id_t ownerId, const RRefId& rrefId, const ForkId& forkId, TypePtr type);
UserRRef(
worker_id_t ownerId,
const RRefId& rrefId,
const ForkId& forkId,
TypePtr type);
inline bool isOwner() const override {
return false;
@ -274,12 +277,15 @@ class TORCH_API OwnerRRef final : public RRef {
OwnerRRef(worker_id_t ownerId, const RRefId& rrefId, TypePtr type)
: OwnerRRef(ownerId, rrefId, type, {}) {}
OwnerRRef(worker_id_t ownerId, const RRefId& rrefId, TypePtr type, c10::optional<IValue> value)
OwnerRRef(
worker_id_t ownerId,
const RRefId& rrefId,
TypePtr type,
c10::optional<IValue> value)
: RRef(ownerId, rrefId, std::move(type)) {
value_ = std::move(value);
}
inline bool isOwner() const override {
return true;
}