From 0170d36c42067809c97cf2adb4d984aefaf8b5d3 Mon Sep 17 00:00:00 2001 From: Qiao Longfei Date: Mon, 15 Oct 2018 14:17:03 +0800 Subject: [PATCH] fix a bug --- paddle/fluid/operators/math/selected_rows_functor_test.cu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddle/fluid/operators/math/selected_rows_functor_test.cu b/paddle/fluid/operators/math/selected_rows_functor_test.cu index c5a23630bb..93e55e88ca 100644 --- a/paddle/fluid/operators/math/selected_rows_functor_test.cu +++ b/paddle/fluid/operators/math/selected_rows_functor_test.cu @@ -288,7 +288,7 @@ TEST(selected_rows_functor, gpu_merge_add) { merge_add_functor(ctx, inputs, output.get()); 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(); EXPECT_EQ(output->height(), height); -- GitLab