Browse Source

!1090 add FUSION_TENSOR_SIZE

From: @lichun30
Reviewed-by: @riverho,@wqtshg,@ji_chen
Signed-off-by: @ji_chen
tags/v1.2.0
mindspore-ci-bot Gitee 3 years ago
parent
commit
4a476f7eb8
2 changed files with 4 additions and 1 deletions
  1. +3
    -0
      inc/external/ge/ge_api_types.h
  2. +1
    -1
      metadef

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

@@ -219,6 +219,9 @@ const std::string HCOM_PARALLEL = "ge.hcomParallel";
// configure whether to use dynamic batch size // configure whether to use dynamic batch size
const char *const kDynamicBatchSize = "ge.dynamicBatchSize"; const char *const kDynamicBatchSize = "ge.dynamicBatchSize";


// configure threshold of fusion data size for communication op
const std::string FUSION_TENSOR_SIZE = "ge.fusionTensorSize";

const std::string INPUT_SHAPE = "ge.inputShape"; const std::string INPUT_SHAPE = "ge.inputShape";


const std::string DYNAMIC_NODE_TYPE = "ge.dynamicNodeType"; const std::string DYNAMIC_NODE_TYPE = "ge.dynamicNodeType";


+ 1
- 1
metadef

@@ -1 +1 @@
Subproject commit 8c5be2db907c26fcd9e7ffdd27f631302140bc2a
Subproject commit e470379501af3042ee5f7e2fa3be338a594fabb3

Loading…
Cancel
Save