Browse Source

fix_auto_parallel_dynamic_shape_map_set

tags/v1.1.0
yao_yf 5 years ago
parent
commit
73dbc8e7ae
1 changed files with 4 additions and 4 deletions
  1. +4
    -4
      mindspore/ccsrc/frontend/parallel/tensor_layout/redistribution_layout_transfer.cc

+ 4
- 4
mindspore/ccsrc/frontend/parallel/tensor_layout/redistribution_layout_transfer.cc View File

@@ -25,10 +25,10 @@ Status RedistributionLayoutTransfer::CheckValidTransfer() {
Shape from_shape = from_in_.tensor_shape().array();
if (std::find(from_shape.begin(), from_shape.end(), -1) != from_shape.end()) {
is_dynamic_shape_ = true;
bool not_all_repeat = std::any_of(from_in_.tensor_map().array().begin(), from_in_.tensor_map().array().end(),
[](int64_t i) { return i != -1; }) ||
std::any_of(to_in_.tensor_map().array().begin(), to_in_.tensor_map().array().end(),
[](int64_t i) { return i != -1; });
Shape from_map = from_in_.tensor_map().array();
Shape to_map = to_in_.tensor_map().array();
bool not_all_repeat = std::any_of(from_map.begin(), from_map.end(), [](int64_t i) { return i != -1; }) ||
std::any_of(to_map.begin(), to_map.end(), [](int64_t i) { return i != -1; });
if (from_in_ != to_in_ && not_all_repeat) {
MS_LOG(ERROR) << "In dynamic shape scene, the from_tensor_shape should be equal to to_tensor_shape";
MS_LOG(ERROR) << "from_in layout" << from_in_.ToString();


Loading…
Cancel
Save