diff --git a/paddle/fluid/operators/checkpoint_op_test.cc b/paddle/fluid/operators/checkpoint_op_test.cc index 1445d9f9acffc9b104a6ee540b7e42b41df0b0f6..bea44b35cada2996a4c8aee0352c349beddf9c2b 100644 --- a/paddle/fluid/operators/checkpoint_op_test.cc +++ b/paddle/fluid/operators/checkpoint_op_test.cc @@ -58,7 +58,7 @@ TEST(CheckpointLoadOp, CPU) { paddle::framework::AttributeMap attrs; attrs.insert({"dir", std::string("ckpt")}); - auto save_op = - paddle::framework::OpRegistry::CreateOp("checkpoint_load", {}, {}, attrs); + auto save_op = paddle::framework::OpRegistry::CreateOp( + "checkpoint_load", {{"X", {"test_var"}}}, {}, attrs); save_op->Run(scope, place); }