register not equal distributed op

10312620 · lichenever · 高东海 · 65cd9d48 · 10312620 · 10312620
Showing with 2 addition and 0 deletion

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

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

未找到文件。
--- 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,