diff --git a/paddle/operators/send_recv_op_benchmark.cc b/paddle/operators/send_recv_op_benchmark.cc index 69131989f080e5f6bb302943a7f2386d577160d6..9fb38e5d6cd5356bd10cd79bedb560fb4aded316 100644 --- a/paddle/operators/send_recv_op_benchmark.cc +++ b/paddle/operators/send_recv_op_benchmark.cc @@ -30,8 +30,9 @@ USE_OP(sum); // global for simplicity. std::unique_ptr recv_op; -int benchmark_count = 100000; -int mat_size = 10; +int benchmark_count = 1000; +// FIXME(typhoonzero): protobuf message size limits the maximum tensor size +int mat_size = 512; void InitTensorsInScope(paddle::framework::Scope &scope, paddle::platform::CPUPlace &place) { @@ -47,7 +48,7 @@ void InitTensorsInScope(paddle::framework::Scope &scope, auto out_var = scope.Var("Out"); auto out_tensor = out_var->GetMutable(); out_tensor->Resize({mat_size, mat_size}); - tensor->mutable_data(place); // allocate + out_tensor->mutable_data(place); // allocate } void AddOp(const std::string &type,