Browse Source

register not equal distributed op

tags/v0.3.0-alpha
lichenever chang zherui 5 years ago
parent
commit
10312620d2
3 changed files with 3 additions and 1 deletions
  1. +1
    -0
      mindspore/ccsrc/parallel/ops_info/ops_utils.h
  2. +1
    -0
      mindspore/ccsrc/parallel/step_auto_parallel.cc
  3. +1
    -1
      mindspore/ccsrc/parallel/step_parallel.cc

+ 1
- 0
mindspore/ccsrc/parallel/ops_info/ops_utils.h View File

@@ -180,6 +180,7 @@ constexpr char SIGMOID[] = "Sigmoid";
constexpr char POW[] = "Pow";
constexpr char MAXIMUM[] = "Maximum";
constexpr char EQUAL[] = "Equal";
constexpr char NOT_EQUAL[] = "NotEqual";
constexpr char LOGICALNOT[] = "LogicalNot";
constexpr char GATHERV2[] = "GatherV2";
constexpr char STRIDEDSLICE[] = "StridedSlice";


+ 1
- 0
mindspore/ccsrc/parallel/step_auto_parallel.cc View File

@@ -95,6 +95,7 @@ std::vector<std::string> splittable_op_ = {MATMUL,
POW,
MAXIMUM,
EQUAL,
NOT_EQUAL,
LOGICALNOT,
GATHERV2,
STRIDEDSLICE,


+ 1
- 1
mindspore/ccsrc/parallel/step_parallel.cc View File

@@ -1226,7 +1226,7 @@ void CoverSliceShape(const FuncGraphPtr& root) {
MS_LOG(INFO) << "Parameter " << parameter->ToString() << " don't need to set parallel shape";
} else {
SetParallelShape(parameter, res);
MS_LOG(DEBUG) << "Parameter " << parameter->ToString() << " shape " << parameter->Shape()->ToString();
MS_LOG(DEBUG) << "Parameter " << parameter->ToString() << " shape " << parameter->Shape()->ToString();
}
}
g_RefMap.clear();


Loading…
Cancel
Save