diff --git a/paddle/fluid/operators/distributed_ops/recv_op.cc b/paddle/fluid/operators/distributed_ops/recv_op.cc index a0185d66f0b9fec4444668fd193763af8cdaa8f1..bcb16ff2e576fc26f9bcd35df02be301d79da379 100644 --- a/paddle/fluid/operators/distributed_ops/recv_op.cc +++ b/paddle/fluid/operators/distributed_ops/recv_op.cc @@ -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(); recv_functor(outs[0], recv_varnames, epmap, exe_ctx, scope); } else { diff --git a/paddle/fluid/operators/distributed_ops/send_op.cc b/paddle/fluid/operators/distributed_ops/send_op.cc index 0f0ad6b8f995bebe9676be80d9947d07eb8e8b1c..801909e2c063e6d6e032fe82536fe9b920dc70b3 100644 --- a/paddle/fluid/operators/distributed_ops/send_op.cc +++ b/paddle/fluid/operators/distributed_ops/send_op.cc @@ -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(); send_functor(ins[0], send_varnames, epmap, height_sections, exe_ctx, scope, static_cast(sync_send));