From 0bd80af1bbd376deca7aafa45f1c4bfceeca46b5 Mon Sep 17 00:00:00 2001 From: "gengchao4@huawei.com" Date: Wed, 14 Jul 2021 20:40:26 +0800 Subject: [PATCH] add support for mds --- ge/graph/passes/mds_kernels/mds_utils.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ge/graph/passes/mds_kernels/mds_utils.h b/ge/graph/passes/mds_kernels/mds_utils.h index d5d9d430..f5199b14 100644 --- a/ge/graph/passes/mds_kernels/mds_utils.h +++ b/ge/graph/passes/mds_kernels/mds_utils.h @@ -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;