diff --git a/paddle/fluid/operators/math/selected_rows_functor_test.cu b/paddle/fluid/operators/math/selected_rows_functor_test.cu index c5a23630bb190d89a1f2f1c9307864ee6b5d96a8..93e55e88ca349fd96925456f135aa5db1aa37fc5 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);