add X to test

7 years ago · 955c79340c
parent 744e95d305
commit 955c79340c
1 changed files with 2 additions and 2 deletions
--- 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);
 }