|
|
|
@ -21,6 +21,7 @@ limitations under the License. */
|
|
|
|
|
|
|
|
|
|
#include <future>
|
|
|
|
|
#include "paddle/fluid/operators/detail/grpc_client.h"
|
|
|
|
|
#include "paddle/fluid/platform/profiler.h"
|
|
|
|
|
|
|
|
|
|
namespace paddle {
|
|
|
|
|
namespace operators {
|
|
|
|
@ -59,6 +60,9 @@ class SendOp : public framework::OperatorBase {
|
|
|
|
|
platform::DeviceContextPool& pool = platform::DeviceContextPool::Instance();
|
|
|
|
|
auto& ctx = *pool.Get(place);
|
|
|
|
|
|
|
|
|
|
// For profiling
|
|
|
|
|
platform::RecordEvent record_event(Type(), &ctx);
|
|
|
|
|
|
|
|
|
|
auto client_var_name = Output("RPCClient");
|
|
|
|
|
PADDLE_ENFORCE_NOT_NULL(scope.FindVar(client_var_name),
|
|
|
|
|
"Can not find variable '%s' in the scope.",
|
|
|
|
|