Skip to content

Commit 1ef6cdb

Browse files
committed
move dist codes from operaotrs/detail to operators/distributed
1 parent 4b7ae14 commit 1ef6cdb

39 files changed

+195
-191
lines changed

paddle/fluid/framework/executor.cc

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ limitations under the License. */
2121
#include "paddle/fluid/framework/op_registry.h"
2222
#include "paddle/fluid/framework/reader.h"
2323
#ifdef PADDLE_WITH_DISTRIBUTE
24-
#include "paddle/fluid/operators/detail/grpc_client.h"
24+
#include "paddle/fluid/operators/distributed/grpc_client.h"
2525
#endif
2626
#include "paddle/fluid/platform/place.h"
2727
#include "paddle/fluid/platform/profiler.h"
@@ -49,8 +49,8 @@ Executor::Executor(const platform::Place& place) : place_(place) {}
4949

5050
#ifdef PADDLE_WITH_DISTRIBUTE
5151
void Executor::Complete() {
52-
::paddle::operators::detail::RPCClient::GetInstance<
53-
::paddle::operators::detail::GRPCClient>()
52+
::paddle::operators::distributed::RPCClient::GetInstance<
53+
::paddle::operators::distributed::GRPCClient>()
5454
->SendComplete();
5555
}
5656
#endif

paddle/fluid/operators/CMakeLists.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -184,8 +184,8 @@ else()
184184
set(DEPS_OPS ${DEPS_OPS} nccl_op)
185185
endif()
186186

187-
add_subdirectory(detail)
188187
if(WITH_DISTRIBUTE)
188+
add_subdirectory(distributed)
189189

190190
set(DISTRIBUTE_DEPS "")
191191
if(WITH_GRPC)

paddle/fluid/operators/detail/macros.h

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -15,13 +15,13 @@
1515
#pragma once
1616

1717
#ifdef PADDLE_WITH_GRPC
18-
#include "paddle/fluid/operators/detail/grpc_client.h"
19-
#include "paddle/fluid/operators/detail/grpc_server.h"
20-
#define RPCSERVER_T detail::AsyncGRPCServer
21-
#define RPCCLIENT_T detail::GRPCClient
18+
#include "paddle/fluid/operators/distributed/grpc_client.h"
19+
#include "paddle/fluid/operators/distributed/grpc_server.h"
20+
#define RPCSERVER_T distributed::AsyncGRPCServer
21+
#define RPCCLIENT_T distributed::GRPCClient
2222
#else
23-
#include "paddle/fluid/operators/detail/brpc_client.h"
24-
#include "paddle/fluid/operators/detail/brpc_server.h"
25-
#define RPCSERVER_T detail::AsyncBRPCServer
26-
#define RPCCLIENT_T detail::BRPCClient
23+
#include "paddle/fluid/operators/distributed/brpc_client.h"
24+
#include "paddle/fluid/operators/distributed/brpc_server.h"
25+
#define RPCSERVER_T distributed::AsyncBRPCServer
26+
#define RPCCLIENT_T distributed::BRPCClient
2727
#endif

paddle/fluid/operators/detail/CMakeLists.txt renamed to paddle/fluid/operators/distributed/CMakeLists.txt

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,3 @@
1-
if(NOT WITH_DISTRIBUTE)
2-
return()
3-
endif()
4-
5-
61
if(WITH_GRPC)
72
grpc_library(sendrecvop_grpc SRCS bytebuffer_stream.cc sendrecvop_utils.cc grpc_client.cc
83
request_handler_impl.cc rpc_client.cc rpc_server.cc grpc_server.cc variable_response.cc PROTO send_recv.proto DEPS lod_tensor

paddle/fluid/operators/detail/brpc_client.cc renamed to paddle/fluid/operators/distributed/brpc_client.cc

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -12,12 +12,12 @@
1212
// See the License for the specific language governing permissions and
1313
// limitations under the License.
1414

15-
#include "paddle/fluid/operators/detail/brpc_client.h"
15+
#include "paddle/fluid/operators/distributed/brpc_client.h"
1616
#include "paddle/fluid/framework/threadpool.h"
1717

1818
namespace paddle {
1919
namespace operators {
20-
namespace detail {
20+
namespace distributed {
2121

2222
DEFINE_int32(brpc_channel_num, 24,
2323
"Number of channels to send requests connected to one server");
@@ -175,6 +175,6 @@ ChannelQueuePtr BRPCClient::GetChannel(const std::string& ep) {
175175
return q;
176176
}
177177

178-
} // namespace detail
178+
} // namespace distributed
179179
} // namespace operators
180180
} // namespace paddle

paddle/fluid/operators/detail/brpc_client.h renamed to paddle/fluid/operators/distributed/brpc_client.h

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -31,13 +31,13 @@ limitations under the License. */
3131
#include "paddle/fluid/framework/lod_tensor.h"
3232
#include "paddle/fluid/framework/scope.h"
3333
#include "paddle/fluid/framework/selected_rows.h"
34-
#include "paddle/fluid/operators/detail/rpc_client.h"
35-
#include "paddle/fluid/operators/detail/send_recv.pb.h"
34+
#include "paddle/fluid/operators/distributed/rpc_client.h"
35+
#include "paddle/fluid/operators/distributed/send_recv.pb.h"
3636
#include "paddle/fluid/platform/macros.h" // for DISABLE_COPY_AND_ASSIGN
3737

3838
namespace paddle {
3939
namespace operators {
40-
namespace detail {
40+
namespace distributed {
4141

4242
struct ChannelContext {
4343
brpc::Channel channel;
@@ -95,6 +95,6 @@ class BRPCClient : public RPCClient {
9595
DISABLE_COPY_AND_ASSIGN(BRPCClient);
9696
};
9797

98-
} // namespace detail
98+
} // namespace distributed
9999
} // namespace operators
100100
} // namespace paddle

paddle/fluid/operators/detail/brpc_server.cc renamed to paddle/fluid/operators/distributed/brpc_server.cc

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -12,13 +12,13 @@
1212
// See the License for the specific language governing permissions and
1313
// limitations under the License.
1414

15-
#include "paddle/fluid/operators/detail/brpc_server.h"
16-
#include "paddle/fluid/operators/detail/request_handler.h"
15+
#include "paddle/fluid/operators/distributed/brpc_server.h"
16+
#include "paddle/fluid/operators/distributed/request_handler.h"
1717

1818
namespace sendrecv {
1919

2020
typedef std::unordered_map<std::string,
21-
paddle::operators::detail::RequestHandler*>
21+
paddle::operators::distributed::RequestHandler*>
2222
HandlerMap;
2323

2424
class BRPCServiceImpl : public SendRecvService {
@@ -27,17 +27,17 @@ class BRPCServiceImpl : public SendRecvService {
2727
: request_send_h_(nullptr),
2828
request_get_h_(nullptr),
2929
request_prefetch_h_(nullptr) {
30-
auto it = rpc_call_map.find(paddle::operators::detail::kRequestSend);
30+
auto it = rpc_call_map.find(paddle::operators::distributed::kRequestSend);
3131
if (it != rpc_call_map.end()) {
3232
request_send_h_ = it->second;
3333
}
3434

35-
it = rpc_call_map.find(paddle::operators::detail::kRequestSend);
35+
it = rpc_call_map.find(paddle::operators::distributed::kRequestSend);
3636
if (it != rpc_call_map.end()) {
3737
request_get_h_ = it->second;
3838
}
3939

40-
it = rpc_call_map.find(paddle::operators::detail::kRequestPrefetch);
40+
it = rpc_call_map.find(paddle::operators::distributed::kRequestPrefetch);
4141
if (it != rpc_call_map.end()) {
4242
request_prefetch_h_ = it->second;
4343
}
@@ -88,15 +88,15 @@ class BRPCServiceImpl : public SendRecvService {
8888
}
8989

9090
private:
91-
paddle::operators::detail::RequestHandler* request_send_h_;
92-
paddle::operators::detail::RequestHandler* request_get_h_;
93-
paddle::operators::detail::RequestHandler* request_prefetch_h_;
91+
paddle::operators::distributed::RequestHandler* request_send_h_;
92+
paddle::operators::distributed::RequestHandler* request_get_h_;
93+
paddle::operators::distributed::RequestHandler* request_prefetch_h_;
9494
};
9595
} // namespace sendrecv
9696

9797
namespace paddle {
9898
namespace operators {
99-
namespace detail {
99+
namespace distributed {
100100

101101
void AsyncBRPCServer::StartServer() {
102102
// Instance of your service.
@@ -139,6 +139,6 @@ void AsyncBRPCServer::WaitServerReady() {
139139
VLOG(3) << "AsyncGRPCServer WaitSeverReady";
140140
}
141141

142-
}; // namespace detail
142+
}; // namespace distributed
143143
}; // namespace operators
144144
}; // namespace paddle

paddle/fluid/operators/detail/brpc_server.h renamed to paddle/fluid/operators/distributed/brpc_server.h

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -19,12 +19,12 @@ limitations under the License. */
1919
#include <string>
2020

2121
#include "brpc/server.h"
22-
#include "paddle/fluid/operators/detail/rpc_server.h"
23-
#include "paddle/fluid/operators/detail/send_recv.pb.h"
22+
#include "paddle/fluid/operators/distributed/rpc_server.h"
23+
#include "paddle/fluid/operators/distributed/send_recv.pb.h"
2424

2525
namespace paddle {
2626
namespace operators {
27-
namespace detail {
27+
namespace distributed {
2828

2929
class AsyncBRPCServer final : public RPCServer {
3030
public:
@@ -48,6 +48,6 @@ class AsyncBRPCServer final : public RPCServer {
4848
int ready_;
4949
};
5050

51-
}; // namespace detail
51+
}; // namespace distributed
5252
}; // namespace operators
5353
}; // namespace paddle

paddle/fluid/operators/detail/bytebuffer_stream.cc renamed to paddle/fluid/operators/distributed/bytebuffer_stream.cc

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -17,11 +17,11 @@ limitations under the License. */
1717
// file and did some modifications so that we can send gRPC
1818
// requests without too much copying of the tensor data.
1919

20-
#include "paddle/fluid/operators/detail/bytebuffer_stream.h"
20+
#include "paddle/fluid/operators/distributed/bytebuffer_stream.h"
2121

2222
namespace paddle {
2323
namespace operators {
24-
namespace detail {
24+
namespace distributed {
2525

2626
GrpcByteBufferSource::GrpcByteBufferSource() {}
2727

@@ -83,6 +83,6 @@ google::protobuf::int64 GrpcByteBufferSource::ByteCount() const {
8383
return byte_count_;
8484
}
8585

86-
} // namespace detail
86+
} // namespace distributed
8787
} // namespace operators
8888
} // namespace paddle

paddle/fluid/operators/detail/bytebuffer_stream.h renamed to paddle/fluid/operators/distributed/bytebuffer_stream.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -106,7 +106,7 @@ class GrpcBufferReader final
106106

107107
namespace paddle {
108108
namespace operators {
109-
namespace detail {
109+
namespace distributed {
110110
// Source provides a way for a particular RPC implementation to provide
111111
// received data to ParseFrom.
112112
class Source {
@@ -183,6 +183,6 @@ class GrpcByteSource : public Source {
183183
char space_[sizeof(Reader)];
184184
};
185185

186-
} // namespace detail
186+
} // namespace distributed
187187
} // namespace operators
188188
} // namespace paddle

0 commit comments

Comments
 (0)