From ff808021c741e28eec2dd2161d2a4ff13e2047b6 Mon Sep 17 00:00:00 2001 From: lichenever Date: Thu, 2 Apr 2020 15:42:00 +0800 Subject: [PATCH] register not equal distributed op --- mindspore/ccsrc/parallel/ops_info/ops_utils.h | 1 + mindspore/ccsrc/parallel/step_auto_parallel.cc | 1 + mindspore/ccsrc/parallel/step_parallel.cc | 2 +- 3 files changed, 3 insertions(+), 1 deletion(-) diff --git a/mindspore/ccsrc/parallel/ops_info/ops_utils.h b/mindspore/ccsrc/parallel/ops_info/ops_utils.h index 2b8fc0ee3fa..4062847d73f 100644 --- 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"; diff --git a/mindspore/ccsrc/parallel/step_auto_parallel.cc b/mindspore/ccsrc/parallel/step_auto_parallel.cc index b06ec383fa1..a359905494f 100644 --- a/mindspore/ccsrc/parallel/step_auto_parallel.cc +++ b/mindspore/ccsrc/parallel/step_auto_parallel.cc @@ -95,6 +95,7 @@ std::vector splittable_op_ = {MATMUL, POW, MAXIMUM, EQUAL, + NOT_EQUAL, LOGICALNOT, GATHERV2, STRIDEDSLICE, diff --git a/mindspore/ccsrc/parallel/step_parallel.cc b/mindspore/ccsrc/parallel/step_parallel.cc index b79198ffcab..78bec00bcf0 100644 --- a/mindspore/ccsrc/parallel/step_parallel.cc +++ b/mindspore/ccsrc/parallel/step_parallel.cc @@ -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();