Skip to content

P
Paddle

PaddlePaddle / Paddle
1 年多 前同步成功

通知 2302
Star 20931
Fork 5422
P
Paddle
提交 57f666fb 编写于 作者: T typhoonzero
浏览文件
操作

update

上级 c66c65cb
隐藏空白更改
内联 并排
Showing with 4 addition and 3 deletion
paddle/operators/send_recv_op_benchmark.cc
浏览文件 @ 57f666fb
...@@ -30,8 +30,9 @@ USE_OP(sum); ...@@ -30,8 +30,9 @@ USE_OP(sum);
// global for simplicity. // global for simplicity.
std::unique_ptr<paddle::framework::OperatorBase> recv_op; std::unique_ptr<paddle::framework::OperatorBase> recv_op;
int benchmark_count = 100000; int benchmark_count = 1000;
int mat_size = 10; // FIXME(typhoonzero): protobuf message size limits the maximum tensor size
int mat_size = 512;
void InitTensorsInScope(paddle::framework::Scope &scope, void InitTensorsInScope(paddle::framework::Scope &scope,
paddle::platform::CPUPlace &place) { paddle::platform::CPUPlace &place) {
...@@ -47,7 +48,7 @@ void InitTensorsInScope(paddle::framework::Scope &scope, ...@@ -47,7 +48,7 @@ void InitTensorsInScope(paddle::framework::Scope &scope,
auto out_var = scope.Var("Out"); auto out_var = scope.Var("Out");
auto out_tensor = out_var->GetMutable<paddle::framework::LoDTensor>(); auto out_tensor = out_var->GetMutable<paddle::framework::LoDTensor>();
out_tensor->Resize({mat_size, mat_size}); out_tensor->Resize({mat_size, mat_size});
tensor->mutable_data<float>(place); // allocate out_tensor->mutable_data<float>(place); // allocate
} }
void AddOp(const std::string &type, void AddOp(const std::string &type,
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或