prune pass simple test

revert-4814-Add_sequence_project_op
Yang Yang 8 years ago
parent 683ef60d7c
commit b68a95f7f4

@ -145,9 +145,11 @@ std::vector<bool> Executor::Preprocess(const ProgramDesc& pdesc) {
// this op should be executed
should_run.push_back(true);
LOG(INFO) << "Yes " << op_desc.type();
} else {
// this op should NOT be executed
should_run.push_back(false);
LOG(INFO) << "No " << op_desc.type();
}
}

@ -38,9 +38,13 @@ class Executor {
protected:
/* @Brief
* Pruning the graph
*
* @param
* ProgramDesc
*
* @return
* vector<bool> Same size as ops. Indicates whether an op should be run.
*/
std::vector<bool> Preprocess(const ProgramDesc& pdesc);

@ -226,6 +226,7 @@ class ExecutorTesterFeed : public ::testing::Test {
Out->set_parameter("Out");
Out->add_arguments("c");
add_fetch_op("a", 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};

Loading…
Cancel
Save