Skip to content

Commit cbf502e

Browse files
committed
fix error
1 parent 270a87f commit cbf502e

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

paddle/fluid/operators/save_load_op_test.cc

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -125,21 +125,21 @@ TEST(LoadFP16Op, CPU) {
125125
save_op->Run(scope, place);
126126

127127
auto load_var = scope.Var("out_var");
128-
auto target = load_var->GetMutable<paddle::framework::LoDTensor>();
129128
auto load_op = paddle::framework::OpRegistry::CreateOp(
130129
"load", {}, {{"Out", {"out_var"}}}, attrs);
131130
LOG(INFO) << "before load op run";
132131
load_op->Run(scope, place);
133132
LOG(INFO) << "after load op run";
134133

135-
paddle::platform::float16* actual = target->data<paddle::platform::float16>();
134+
auto target = load_var->Get<paddle::framework::LoDTensor>();
135+
paddle::platform::float16* actual = target.data<paddle::platform::float16>();
136136
LOG(INFO) << "after target->data";
137137
for (int64_t i = 0; i < tensor->numel(); ++i) {
138138
EXPECT_EQ(expect[i], static_cast<float>(actual[i]));
139139
}
140140
LOG(INFO) << "after expect equal";
141141

142-
auto& actual_lod = target->lod();
142+
auto& actual_lod = target.lod();
143143
EXPECT_EQ(expect_lod.size(), actual_lod.size());
144144
for (size_t i = 0; i < expect_lod.size(); ++i) {
145145
for (size_t j = 0; j < expect_lod[i].size(); ++j) {

0 commit comments

Comments
 (0)