|
|
@@ -110,6 +110,7 @@ const char *const SAVE_ORIGINAL_MODEL = "ge.saveOriginalModel"; |
|
|
|
const char *const ORIGINAL_MODEL_FILE = "ge.originalModelFile"; |
|
|
|
const char *const INPUT_FP16_NODES = "ge.INPUT_NODES_SET_FP16"; |
|
|
|
const char *const OP_DEBUG_LEVEL = "ge.opDebugLevel"; |
|
|
|
const char *const PERFORMANCE_MODE = "ge.performance_mode"; |
|
|
|
} // namespace configure_option |
|
|
|
// Configure stream num by Session constructor options param, |
|
|
|
// its value should be int32_t type, default value is "1" |
|
|
@@ -314,6 +315,11 @@ const std::string HCOM_MULTI_MODE = "ge.hcomMultiMode"; |
|
|
|
// atc and ir option |
|
|
|
const char *const INPUT_SHAPE_RANGE = "input_shape_range"; |
|
|
|
|
|
|
|
// Configure express high compile performance or high execute performance |
|
|
|
// normal: no need to compile, used saved .o files directly |
|
|
|
// high: need to recompile, high execute performance mode |
|
|
|
const std::string PERFORMANCE_MODE = "ge.performance_mode"; |
|
|
|
|
|
|
|
// Graph run mode |
|
|
|
enum GraphRunMode { PREDICTION = 0, TRAIN }; |
|
|
|
|
|
|
@@ -388,6 +394,7 @@ 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(); |
|
|
|
static const char *const PERFORMANCE_MODE = ge::PERFORMANCE_MODE.c_str(); |
|
|
|
|
|
|
|
// for interface: aclgrphBuildModel |
|
|
|
#ifdef __GNUC__ |
|
|
@@ -412,7 +419,8 @@ const std::set<std::string> ir_builder_suppported_options = {INPUT_FORMAT, |
|
|
|
OP_COMPILER_CACHE_MODE, |
|
|
|
MDL_BANK_PATH, |
|
|
|
OP_BANK_PATH, |
|
|
|
OP_BANK_UPDATE}; |
|
|
|
OP_BANK_UPDATE, |
|
|
|
PERFORMANCE_MODE}; |
|
|
|
|
|
|
|
// for interface: aclgrphParse |
|
|
|
const std::set<std::string> ir_parser_suppported_options = { |
|
|
|