diff --git a/paddle/fluid/operators/distributed_ops/recv_op.cc b/paddle/fluid/operators/distributed_ops/recv_op.cc index a0185d66f0..bcb16ff2e5 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 0f0ad6b8f9..801909e2c0 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));