|
@@ -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 }; |
|
|
|
|
|
|
|
|