!105 register not equal distributed op

Merge pull request !105 from lichen/register_not_equal_distributed_op
5 years ago · ad88c55859
parent cb323eccf0 ff808021c7
commit ad88c55859
3 changed files with 3 additions and 1 deletions
--- a/mindspore/ccsrc/parallel/ops_info/ops_utils.h
+++ b/mindspore/ccsrc/parallel/ops_info/ops_utils.h
@ -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";
--- a/mindspore/ccsrc/parallel/step_auto_parallel.cc
+++ b/mindspore/ccsrc/parallel/step_auto_parallel.cc
@ -95,6 +95,7 @@ std::vector<std::string> splittable_op_ = {MATMUL,
                                           POW,
                                           MAXIMUM,
                                           EQUAL,
+                                           NOT_EQUAL,
                                           LOGICALNOT,
                                           GATHERV2,
                                           STRIDEDSLICE,