Browse Source

add FUSION_TENSOR_SIZE

tags/v1.2.0
lichun 4 years ago
parent
commit
1a5c0ea709
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
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 DYNAMIC_NODE_TYPE = "ge.dynamicNodeType";


+ 1
- 1
metadef

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

Loading…
Cancel
Save