From 0d85b6c0a8ce441033accb66ee028d075465a029 Mon Sep 17 00:00:00 2001 From: wxl Date: Fri, 18 Dec 2020 16:50:48 +0800 Subject: [PATCH] add option for ir --- inc/external/ge/ge_api_types.h | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/inc/external/ge/ge_api_types.h b/inc/external/ge/ge_api_types.h index cce17f93..59e8808b 100644 --- a/inc/external/ge/ge_api_types.h +++ b/inc/external/ge/ge_api_types.h @@ -293,6 +293,7 @@ const std::string MDL_BANK_PATH_FLAG = "ge.mdl_bank_path"; // Configure op bank path const std::string OP_BANK_PATH_FLAG = "ge.op_bank_path"; +const std::string OP_BANK_UPDATE_FLAG = "ge.op_bank_update"; // Graph run mode enum GraphRunMode { PREDICTION = 0, TRAIN }; @@ -366,6 +367,7 @@ static const char *const OP_COMPILER_CACHE_DIR = ge::OP_COMPILER_CACHE_DIR; static const char *const OP_COMPILER_CACHE_MODE = ge::OP_COMPILER_CACHE_MODE; static const char *const MDL_BANK_PATH = ge::MDL_BANK_PATH_FLAG.c_str(); static const char *const OP_BANK_PATH = ge::OP_BANK_PATH_FLAG.c_str(); +static const char *const OP_BANK_UPDATE = ge::OP_BANK_UPDATE_FLAG.c_str(); static const char *const OP_DEBUG_LEVEL = ge::OP_DEBUG_LEVEL.c_str(); // for interface: aclgrphBuildModel @@ -389,7 +391,8 @@ const std::set ir_builder_suppported_options = {INPUT_FORMAT, OP_COMPILER_CACHE_DIR, OP_COMPILER_CACHE_MODE, MDL_BANK_PATH, - OP_BANK_PATH}; + OP_BANK_PATH, + OP_BANK_UPDATE}; // for interface: aclgrphParse const std::set ir_parser_suppported_options = {INPUT_FORMAT,