From 10312620d2d946f95dfdbaad841a3299f376eaaf 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 2b8fc0ee3f..4062847d73 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 b06ec383fa..a359905494 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 b79198ffca..78bec00bcf 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();