pj3fizlut
/
mindspore
forked from OSSInnovation/mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

!2746 [AutoParallel]Remove 32 byte aligned limit for CPU gatherV2 

Merge pull request !2746 from lichen/remove_32byte_limit_for_cpu_gatherv2
This commit is contained in:
mindspore-ci-bot 2020-06-30 20:29:14 +08:00 committed by Gitee
parent 3a89f93cc5 3398c2d74b
commit 2dd5cf14da
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/ccsrc/parallel/ops_info/gather_v2_p_info.cc
View File

@ -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;
}