|
|
|
@ -156,6 +156,7 @@ TEST(SendRecvOp, CPUDense) {
|
|
|
|
|
std::thread server_thread(StartServerNet, false, &initialized);
|
|
|
|
|
while (!initialized) {
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static_cast<paddle::operators::ListenAndServOp *>(listen_and_serv_op.get())
|
|
|
|
|
->WaitServerReady();
|
|
|
|
|
|
|
|
|
@ -175,9 +176,10 @@ TEST(SendRecvOp, CPUDense) {
|
|
|
|
|
std::string endpoint = paddle::string::Sprintf("127.0.0.1:%d", selected_port);
|
|
|
|
|
attrs.insert({"endpoints", std::vector<std::string>({endpoint})});
|
|
|
|
|
attrs.insert({"epmap", std::vector<std::string>({endpoint})});
|
|
|
|
|
auto send_op = f::OpRegistry::CreateOp(
|
|
|
|
|
"send", {{"X", {"x1"}}},
|
|
|
|
|
{{"Out", {"Out"}}, attrs);
|
|
|
|
|
const f::VariableNameMap &inputs = {{"X", {"x1"}}};
|
|
|
|
|
const f::VariableNameMap &outputs = {{"Out", {"Out"}}};
|
|
|
|
|
|
|
|
|
|
auto send_op = f::OpRegistry::CreateOp("send", inputs, outputs, attrs);
|
|
|
|
|
send_op->Run(scope, place);
|
|
|
|
|
|
|
|
|
|
auto in_var = scope.Var("x1");
|
|
|
|
@ -217,8 +219,7 @@ TEST(SendRecvOp, CPUSparse) {
|
|
|
|
|
scope.Var("RPC_CLIENT_VAR");
|
|
|
|
|
f::AttributeMap attrs;
|
|
|
|
|
selected_port = listen_and_serv_op_ptr->GetSelectedPort();
|
|
|
|
|
std::string endpoint =
|
|
|
|
|
paddle::string::Sprintf("127.0.0.1:%d", selected_port);
|
|
|
|
|
std::string endpoint = paddle::string::Sprintf("127.0.0.1:%d", selected_port);
|
|
|
|
|
attrs.insert({"endpoints", std::vector<std::string>({endpoint})});
|
|
|
|
|
attrs.insert({"epmap", std::vector<std::string>({endpoint})});
|
|
|
|
|
auto send_op = f::OpRegistry::CreateOp("send", {{"X", {"x1"}}},
|
|
|
|
|