Browse Source

add multi mode config

pull/277/head
wangxiaotian22 4 years ago
parent
commit
1852d7fefc
1 changed files with 5 additions and 0 deletions
  1. +5
    -0
      inc/external/ge/ge_api_types.h

+ 5
- 0
inc/external/ge/ge_api_types.h View File

@@ -245,6 +245,11 @@ const std::string INPUT_FP16_NODES = "ge.INPUT_NODES_SET_FP16";
// 0: close debug; 1: open TBE compiler; 2: open ccec compiler // 0: close debug; 1: open TBE compiler; 2: open ccec compiler
const std::string OP_DEBUG_LEVEL = "ge.opDebugLevel"; const std::string OP_DEBUG_LEVEL = "ge.opDebugLevel";


// Configure for fix hcombroadcast format.
// when config model multi, broadcast format should be fixed
// 0: data multi; 1: model multi;
const std::string HCOM_MULTI_MODE = "ge.hcomMultiMode";

// Graph run mode // Graph run mode
enum GraphRunMode { PREDICTION = 0, TRAIN }; enum GraphRunMode { PREDICTION = 0, TRAIN };




Loading…
Cancel
Save