|
|
|
@ -201,6 +201,10 @@ DEFINE_string(op_compiler_cache_dir, "", "Optional; the path to cache operator c
|
|
|
|
|
|
|
|
|
|
DEFINE_string(op_compiler_cache_mode, "", "Optional; choose the operator compiler cache mode");
|
|
|
|
|
|
|
|
|
|
DEFINE_string(mdl_bank_path, "", "Optional; model bank path");
|
|
|
|
|
|
|
|
|
|
DEFINE_string(op_bank_path, "", "Optional; op bank path");
|
|
|
|
|
|
|
|
|
|
class GFlagUtils {
|
|
|
|
|
public:
|
|
|
|
|
/**
|
|
|
|
@ -1017,6 +1021,8 @@ static void SetEnvForSingleOp(std::map<string, string> &options) {
|
|
|
|
|
options.emplace(ge::DEBUG_DIR, FLAGS_debug_dir);
|
|
|
|
|
options.emplace(ge::OP_COMPILER_CACHE_DIR, FLAGS_op_compiler_cache_dir);
|
|
|
|
|
options.emplace(ge::OP_COMPILER_CACHE_MODE, FLAGS_op_compiler_cache_mode);
|
|
|
|
|
options.emplace(ge::MDL_BANK_PATH_FLAG, FLAGS_mdl_bank_path);
|
|
|
|
|
options.emplace(ge::OP_BANK_PATH_FLAG, FLAGS_op_bank_path);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
domi::Status GenerateSingleOp(const std::string& json_file_path) {
|
|
|
|
@ -1170,6 +1176,10 @@ domi::Status GenerateOmModel() {
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
options.insert(std::pair<string, string>(string(ge::OP_DEBUG_LEVEL), to_string(FLAGS_op_debug_level)));
|
|
|
|
|
|
|
|
|
|
options.insert(std::pair<string, string>(string(ge::MDL_BANK_PATH_FLAG), FLAGS_mdl_bank_path));
|
|
|
|
|
|
|
|
|
|
options.insert(std::pair<string, string>(string(ge::OP_BANK_PATH_FLAG), FLAGS_op_bank_path));
|
|
|
|
|
// set enable scope fusion passes
|
|
|
|
|
SetEnableScopeFusionPasses(FLAGS_enable_scope_fusion_passes);
|
|
|
|
|
// print atc option map
|
|
|
|
|