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

!105 register not equal distributed op 

Merge pull request !105 from lichen/register_not_equal_distributed_op
This commit is contained in:
mindspore-ci-bot 2020-04-02 21:34:34 +08:00 committed by Gitee
parent cb323eccf0 ff808021c7
commit ad88c55859
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 POW[] = "Pow";
constexpr char MAXIMUM[] = "Maximum"; constexpr char MAXIMUM[] = "Maximum";
constexpr char EQUAL[] = "Equal"; constexpr char EQUAL[] = "Equal";
constexpr char NOT_EQUAL[] = "NotEqual";
constexpr char LOGICALNOT[] = "LogicalNot"; constexpr char LOGICALNOT[] = "LogicalNot";
constexpr char GATHERV2[] = "GatherV2"; constexpr char GATHERV2[] = "GatherV2";
constexpr char STRIDEDSLICE[] = "StridedSlice"; 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, POW,
MAXIMUM, MAXIMUM,
EQUAL, EQUAL,
NOT_EQUAL,
LOGICALNOT, LOGICALNOT,
GATHERV2, GATHERV2,
STRIDEDSLICE, 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"; MS_LOG(INFO) << "Parameter " << parameter->ToString() << " don't need to set parallel shape";
} else { } else {
SetParallelShape(parameter, res); 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(); g_RefMap.clear();