From 3398c2d74bb69311ab6ebe75ccf64869638f79c8 Mon Sep 17 00:00:00 2001 From: lichenever Date: Tue, 30 Jun 2020 11:53:12 +0800 Subject: [PATCH] remove_32byte_limit_for_cpu_gatherv2 --- mindspore/ccsrc/parallel/ops_info/gather_v2_p_info.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mindspore/ccsrc/parallel/ops_info/gather_v2_p_info.cc b/mindspore/ccsrc/parallel/ops_info/gather_v2_p_info.cc index 7a16aeafcb..9fb8df0883 100644 --- a/mindspore/ccsrc/parallel/ops_info/gather_v2_p_info.cc +++ b/mindspore/ccsrc/parallel/ops_info/gather_v2_p_info.cc @@ -79,8 +79,8 @@ Status GatherV2PInfo::CheckStrategy(const StrategyPtr &strategy) { auto param_shape = inputs_shape_.at(0); auto param_strategy = strategy->GetInputDim().at(0); auto slice_shape = param_shape.at(param_shape.size() - 1) / param_strategy.at(param_strategy.size() - 1); - if (slice_shape % 8 != 0 && slice_shape != 1) { - MS_LOG(DEBUG) << name_ << ": Last dim of param slice shape need 32Byte aligned."; + if ((target_ != CPU) && (slice_shape % 8 != 0) && (slice_shape != 1)) { + MS_LOG(ERROR) << name_ << ": Last dim of param slice shape need 32Byte aligned."; return FAILED; }