diff --git a/paddle/fluid/operators/save_load_op_test.cc b/paddle/fluid/operators/save_load_op_test.cc index 6a5ae01347569b4bd55df9f2e718c1b9842b894d..9c15461305c45363151cd70fb3f9614e860660b1 100644 --- a/paddle/fluid/operators/save_load_op_test.cc +++ b/paddle/fluid/operators/save_load_op_test.cc @@ -125,21 +125,21 @@ TEST(LoadFP16Op, CPU) { save_op->Run(scope, place); auto load_var = scope.Var("out_var"); - auto target = load_var->GetMutable(); auto load_op = paddle::framework::OpRegistry::CreateOp( "load", {}, {{"Out", {"out_var"}}}, attrs); LOG(INFO) << "before load op run"; load_op->Run(scope, place); LOG(INFO) << "after load op run"; - paddle::platform::float16* actual = target->data(); + auto target = load_var->Get(); + paddle::platform::float16* actual = target.data(); LOG(INFO) << "after target->data"; for (int64_t i = 0; i < tensor->numel(); ++i) { EXPECT_EQ(expect[i], static_cast(actual[i])); } LOG(INFO) << "after expect equal"; - auto& actual_lod = target->lod(); + auto& actual_lod = target.lod(); EXPECT_EQ(expect_lod.size(), actual_lod.size()); for (size_t i = 0; i < expect_lod.size(); ++i) { for (size_t j = 0; j < expect_lod[i].size(); ++j) {