提交 0170d36c 编写于 作者: Q Qiao Longfei

fix a bug

上级 86e2e686
...@@ -288,7 +288,7 @@ TEST(selected_rows_functor, gpu_merge_add) { ...@@ -288,7 +288,7 @@ TEST(selected_rows_functor, gpu_merge_add) {
merge_add_functor(ctx, inputs, output.get()); merge_add_functor(ctx, inputs, output.get());
paddle::framework::Tensor output_cpu; paddle::framework::Tensor output_cpu;
paddle::framework::TensorCopy(output.value(), cpu_place, ctx, &output_cpu); paddle::framework::TensorCopy(output->value(), cpu_place, ctx, &output_cpu);
ctx.Wait(); ctx.Wait();
EXPECT_EQ(output->height(), height); EXPECT_EQ(output->height(), height);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册