fix compile

mixed_precision_init
Qiao Longfei 6 years ago
parent 49f2f4f91d
commit 02425b2f64

@ -57,7 +57,7 @@ class RecvOp : public framework::OperatorBase {
platform::DeviceContextPool &pool =
platform::DeviceContextPool::Instance();
auto *dev_ctx = pool.Get(place);
auto exe_ctx = framework::ExecutionContext(*this, scope, *dev_ctx, ctx);
auto exe_ctx = framework::ExecutionContext(*this, scope, *dev_ctx, ctx, nullptr);
auto recv_functor = distributed::ParameterRecv<float>();
recv_functor(outs[0], recv_varnames, epmap, exe_ctx, scope);
} else {

@ -50,7 +50,7 @@ class SendOp : public framework::OperatorBase {
platform::DeviceContextPool& pool =
platform::DeviceContextPool::Instance();
auto* dev_ctx = pool.Get(place);
auto exe_ctx = framework::ExecutionContext(*this, scope, *dev_ctx, ctx);
auto exe_ctx = framework::ExecutionContext(*this, scope, *dev_ctx, ctx, nullptr);
auto send_functor = distributed::ParameterSend<float>();
send_functor(ins[0], send_varnames, epmap, height_sections, exe_ctx,
scope, static_cast<bool>(sync_send));

Loading…
Cancel
Save