提交 b68a95f7 编写于 作者: Y Yang Yang

prune pass simple test

上级 683ef60d
...@@ -145,9 +145,11 @@ std::vector<bool> Executor::Preprocess(const ProgramDesc& pdesc) { ...@@ -145,9 +145,11 @@ std::vector<bool> Executor::Preprocess(const ProgramDesc& pdesc) {
// this op should be executed // this op should be executed
should_run.push_back(true); should_run.push_back(true);
LOG(INFO) << "Yes " << op_desc.type();
} else { } else {
// this op should NOT be executed // this op should NOT be executed
should_run.push_back(false); should_run.push_back(false);
LOG(INFO) << "No " << op_desc.type();
} }
} }
......
...@@ -38,9 +38,13 @@ class Executor { ...@@ -38,9 +38,13 @@ class Executor {
protected: protected:
/* @Brief /* @Brief
* Pruning the graph
* *
* @param * @param
* ProgramDesc * ProgramDesc
*
* @return
* vector<bool> Same size as ops. Indicates whether an op should be run.
*/ */
std::vector<bool> Preprocess(const ProgramDesc& pdesc); std::vector<bool> Preprocess(const ProgramDesc& pdesc);
......
...@@ -226,6 +226,7 @@ class ExecutorTesterFeed : public ::testing::Test { ...@@ -226,6 +226,7 @@ class ExecutorTesterFeed : public ::testing::Test {
Out->set_parameter("Out"); Out->set_parameter("Out");
Out->add_arguments("c"); Out->add_arguments("c");
add_fetch_op("a", dim, 0, root_block);
add_fetch_op("c", dim, 0, root_block); add_fetch_op("c", dim, 0, root_block);
std::vector<float> vec1 = {1.0, 2.0, 3.0, 4.0, 5.0, 6.0}; std::vector<float> vec1 = {1.0, 2.0, 3.0, 4.0, 5.0, 6.0};
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册