| @@ -53,6 +53,8 @@ const std::string IR_OPTION_ENABLE_COMPRESS_WEIGHT_DEFAULT = "false"; | |||||
| const std::string kInputShape = "input_shape"; | const std::string kInputShape = "input_shape"; | ||||
| const std::string kInputFormat = "input_format"; | const std::string kInputFormat = "input_format"; | ||||
| const std::string kReUseMemEnable = "1"; | |||||
| const std::string kReUseMemDisEnable = "0"; | |||||
| } // namespace | } // namespace | ||||
| static graphStatus CheckGlobalOptions(std::map<std::string, std::string> &global_options) { | static graphStatus CheckGlobalOptions(std::map<std::string, std::string> &global_options) { | ||||
| @@ -313,6 +315,12 @@ graphStatus Impl::CheckOptions(const std::map<std::string, std::string> &options | |||||
| return GRAPH_PARAM_INVALID; | return GRAPH_PARAM_INVALID; | ||||
| } | } | ||||
| } | } | ||||
| // Check option EXEC_DISABLE_REUSED_MEMORY | |||||
| it = options_.find(EXEC_DISABLE_REUSED_MEMORY); | |||||
| if (it != options_.end() && it->second != kReUseMemEnable && it->second != kReUseMemDisEnable) { | |||||
| GELOGE(GRAPH_PARAM_INVALID, "option(EXEC_DISABLE_REUSED_MEMORY) value[%s] is invalid ", it->second.c_str()); | |||||
| return GRAPH_PARAM_INVALID; | |||||
| } | |||||
| return GRAPH_SUCCESS; | return GRAPH_SUCCESS; | ||||
| } | } | ||||