提交 955c7934 编写于 作者: T tangwei12

add X to test

上级 744e95d3
...@@ -58,7 +58,7 @@ TEST(CheckpointLoadOp, CPU) { ...@@ -58,7 +58,7 @@ TEST(CheckpointLoadOp, CPU) {
paddle::framework::AttributeMap attrs; paddle::framework::AttributeMap attrs;
attrs.insert({"dir", std::string("ckpt")}); attrs.insert({"dir", std::string("ckpt")});
auto save_op = auto save_op = paddle::framework::OpRegistry::CreateOp(
paddle::framework::OpRegistry::CreateOp("checkpoint_load", {}, {}, attrs); "checkpoint_load", {{"X", {"test_var"}}}, {}, attrs);
save_op->Run(scope, place); save_op->Run(scope, place);
} }
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册