Browse Source

bugfix

tags/v1.2.0
wxl 4 years ago
parent
commit
0dfbbe1c1e
2 changed files with 5 additions and 3 deletions
  1. +4
    -2
      ge/ir_build/atc_ir_common.cc
  2. +1
    -1
      ge/ir_build/atc_ir_common.h

+ 4
- 2
ge/ir_build/atc_ir_common.cc View File

@@ -63,12 +63,14 @@ vector<string> SplitInputShape(const std::string &input_shape) {
}
} // namespace

Status CheckInputFormat(string &input_format) {
Status CheckInputFormat(const string &input_format) {
if (input_format.empty()) {
return ge::SUCCESS;
}
if (!ge::TypeUtils::IsFormatValid(input_format.c_str())) {
GELOGE(ge::PARAM_INVALID, "user input format [%s] is not found!", input_format.c_str());
ErrorManager::GetInstance().ATCReportErrMessage(
"E10001", {"parameter", "value", "reason"}, {"--input_format", FLAGS_input_format, "input format is invalid!"});
GELOGE(ge::PARAM_INVALID, "input format [%s] is invalid!", input_format.c_str());
return ge::PARAM_INVALID;
}
return ge::SUCCESS;


+ 1
- 1
ge/ir_build/atc_ir_common.h View File

@@ -75,7 +75,7 @@ Status CheckInsertOpConfParamValid(const std::string insert_op_conf);
Status CheckDisableReuseMemoryParamValid(const std::string disable_reuse_memory);
Status CheckEnableSingleStreamParamValid(const std::string enable_single_stream);
Status CheckImplmodeParamValid(const std::string &optypelist_for_implmode, std::string &op_select_implmode);
Status CheckInputFormat(string &input_format);
Status CheckInputFormat(const string &input_format);
void PrintOptionMap(std::map<std::string, std::string> &options, std::string tips);
void EraseEndSemicolon(std::string &param);
}


Loading…
Cancel
Save