提交 ea765876 编写于 作者: H huzhiqiang 提交者: GitHub

[FIX OPT] Make opt support models set input (#2918)

上级 ee74f7dc
...@@ -30,6 +30,7 @@ ...@@ -30,6 +30,7 @@
#include "lite/model_parser/compatible_pb.h" #include "lite/model_parser/compatible_pb.h"
#include "lite/model_parser/pb/program_desc.h" #include "lite/model_parser/pb/program_desc.h"
#include "lite/utils/cp_logging.h" #include "lite/utils/cp_logging.h"
#include "lite/utils/io.h"
#include "lite/utils/string.h" #include "lite/utils/string.h"
#include "supported_kernel_op_info.h" // NOLINT #include "supported_kernel_op_info.h" // NOLINT
...@@ -400,6 +401,7 @@ void Main() { ...@@ -400,6 +401,7 @@ void Main() {
return; return;
} }
lite::MkDirRecur(FLAGS_optimize_out);
auto model_dirs = lite::ListDir(FLAGS_model_set_dir, true); auto model_dirs = lite::ListDir(FLAGS_model_set_dir, true);
if (model_dirs.size() == 0) { if (model_dirs.size() == 0) {
LOG(FATAL) << "[" << FLAGS_model_set_dir << "] does not contain any model"; LOG(FATAL) << "[" << FLAGS_model_set_dir << "] does not contain any model";
...@@ -454,7 +456,9 @@ int main(int argc, char** argv) { ...@@ -454,7 +456,9 @@ int main(int argc, char** argv) {
} }
google::ParseCommandLineFlags(&argc, &argv, false); google::ParseCommandLineFlags(&argc, &argv, false);
paddle::lite_api::ParseInputCommand(); paddle::lite_api::ParseInputCommand();
paddle::lite_api::CheckIfModelSupported(); if (FLAGS_model_set_dir == "") {
paddle::lite_api::CheckIfModelSupported();
}
paddle::lite_api::Main(); paddle::lite_api::Main();
return 0; return 0;
} }
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册