From 1852d7fefcd9b7d9789a5cefb5eb0e4cf5957ac7 Mon Sep 17 00:00:00 2001 From: wangxiaotian22 Date: Mon, 9 Nov 2020 16:52:57 +0800 Subject: [PATCH] add multi mode config --- inc/external/ge/ge_api_types.h | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/inc/external/ge/ge_api_types.h b/inc/external/ge/ge_api_types.h index 113b8bc6..1a843b03 100644 --- a/inc/external/ge/ge_api_types.h +++ b/inc/external/ge/ge_api_types.h @@ -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 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 enum GraphRunMode { PREDICTION = 0, TRAIN };