Browse Source

add help info for debug_dir & op_compiler_cache_dir/mode

pull/209/head
wuweikang 4 years ago
parent
commit
82016e82a5
3 changed files with 2222 additions and 140 deletions
  1. +2200
    -133
      CMakeLists.txt
  2. +5
    -1
      ge/offline/main.cc
  3. +17
    -6
      inc/external/ge/ge_api_types.h

+ 2200
- 133
CMakeLists.txt
File diff suppressed because it is too large
View File


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

@@ -299,7 +299,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

@@ -311,12 +311,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: aclgrphBuildInitialize
const std::set<std::string> global_options = {CORE_TYPE,
SOC_VERSION,


Loading…
Cancel
Save