Browse Source

add support for mds

pull/2000/head
gengchao4@huawei.com 3 years ago
parent
commit
0bd80af1bb
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/graph/passes/mds_kernels/mds_utils.h

+ 1
- 1
ge/graph/passes/mds_kernels/mds_utils.h View File

@@ -60,7 +60,7 @@ const int32_t kNchwDimIdxW = 3;
// default die number
const uint32_t kDeployNumber = 2;
enum CutType { kNoCut = 0, kCutN, kCutH, kDynamicCutN, kDynamicCutH, kDynamicCutAll };
enum TensorCutInfo { kNotSupport = 0, kSplitCutSupported, kAnyCutSupported = 4 };
enum TensorCutInfo { kNotSupport = 0, kSplitCutSupported, kAnyCutSupported = 3 };
const int64_t kDefaultFissionFactor = 1;
const int64_t kDefaultRankSize = 1;


Loading…
Cancel
Save