Browse Source

!275 add help info for debug_dir & op_compiler_cache_dir/mode

From: @HW_KK
Reviewed-by: @youui
Signed-off-by:
tags/v1.1.0
mindspore-ci-bot Gitee 3 years ago
parent
commit
4ed251722d
2 changed files with 22 additions and 7 deletions
  1. +5
    -1
      ge/offline/main.cc
  2. +17
    -6
      inc/external/ge/ge_api_types.h

+ 5
- 1
ge/offline/main.cc View File

@@ -300,7 +300,11 @@ class GFlagUtils {
" --save_original_model Control whether to output original model. E.g.: true: output original model\n"
" --log Generate log with level. Support debug, info, warning, error, null\n"
" --dump_mode The switch of dump json with shape, to be used with mode 1. "
"0(default): disable; 1: enable.");
"0(default): disable; 1: enable.\n"
" --debug_dir Set the save path of operator compilation intermediate files. Default value: ./\n"
" --op_compiler_cache_dir Set the save path of operator compilation cache files. Default value: ./\n"
" --op_compiler_cache_mode Set the operator compilation cache mode."
"Options are disable(default), enable and force(force to refresh the cache)");

gflags::ParseCommandLineNonHelpFlags(&argc, &argv, true);
// Using gflags to analyze input parameters


+ 17
- 6
inc/external/ge/ge_api_types.h View File

@@ -316,12 +316,23 @@ static const char *const DEBUG_DIR = ge::DEBUG_DIR;
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;
// for interface: aclgrphBuildModel
const std::set<std::string> ir_builder_suppported_options = {
INPUT_FORMAT, INPUT_SHAPE, OP_NAME_MAP,
DYNAMIC_BATCH_SIZE, DYNAMIC_IMAGE_SIZE, DYNAMIC_DIMS,
INSERT_OP_FILE, PRECISION_MODE, EXEC_DISABLE_REUSED_MEMORY,
AUTO_TUNE_MODE, OUTPUT_TYPE, OUT_NODES,
INPUT_FP16_NODES, LOG_LEVEL};
const std::set<std::string> ir_builder_suppported_options = {INPUT_FORMAT,
INPUT_SHAPE,
OP_NAME_MAP,
DYNAMIC_BATCH_SIZE,
DYNAMIC_IMAGE_SIZE,
DYNAMIC_DIMS,
INSERT_OP_FILE,
PRECISION_MODE,
EXEC_DISABLE_REUSED_MEMORY,
AUTO_TUNE_MODE,
OUTPUT_TYPE,
OUT_NODES,
INPUT_FP16_NODES,
LOG_LEVEL,
DEBUG_DIR,
OP_COMPILER_CACHE_DIR,
OP_COMPILER_CACHE_MODE};

// for interface: aclgrphParse
const std::set<std::string> ir_parser_suppported_options = {INPUT_FORMAT,


Loading…
Cancel
Save