Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

register not equal distributed op

This commit is contained in:
lichenever 2020-04-02 15:42:00 +08:00 committed by chang zherui
parent 65cd9d48fd
commit 10312620d2
3 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
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
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,

2
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();