!6403 fix codex clean bug

Merge pull request !6403 from laiyongqiang/codex2
4 years ago · deb1d2ba96
parent ab75268187 e13beaa7ed
commit deb1d2ba96
1 changed files with 1 additions and 1 deletions
--- a/mindspore/ccsrc/backend/optimizer/mem_reuse/mem_reuse_allocator.cc
+++ b/mindspore/ccsrc/backend/optimizer/mem_reuse/mem_reuse_allocator.cc
@ -204,7 +204,7 @@ void BestFitMemReuse::AssignCommunicationNodeOutputOffset() {
    size_t index = GetTensorIndex(tensor_idx);
    auto tensor_desc = tensor_ptr_list_[index];
    MS_EXCEPTION_IF_NULL(tensor_desc);
-    if (output_index == 0 || output_index == output_ref_indexes.size() - 1) {
+    if (output_index == 0) {
      tensor_desc->size_ += kDefaultMemAlignSize;
    }