This is an automated email from the ASF dual-hosted git repository. wuwei pushed a commit to branch unity in repository https://gitbox.apache.org/repos/asf/tvm.git
commit 6b6419c55f79878d5c79a8c9fbf88796bd490d17 Author: LeiWang <[email protected]> AuthorDate: Wed Jan 3 12:34:03 2024 -0400 merge from unity upstream --- src/tir/transforms/merge_shared_memory_allocations.cc | 12 +----------- 1 file changed, 1 insertion(+), 11 deletions(-) diff --git a/src/tir/transforms/merge_shared_memory_allocations.cc b/src/tir/transforms/merge_shared_memory_allocations.cc index c0e7cf1ebe..207d3c5dac 100644 --- a/src/tir/transforms/merge_shared_memory_allocations.cc +++ b/src/tir/transforms/merge_shared_memory_allocations.cc @@ -473,13 +473,9 @@ class SharedMemoryRewriter : public StmtExprMutator { // - leaf stmt(offset = 0) // - end of scope(offset < 0) // In both cases, we need to handle the kill event correctly - auto is_leaf_alloc = [&](const VarNode* var) { - return seq[i].scope_pair_offset == 0 && - std::find(it->second.gen.begin(), it->second.gen.end(), var) != it->second.gen.end(); - }; if (it != event_map_.end() && seq[i].scope_pair_offset <= 0) { for (const VarNode* var : it->second.kill) { - if (!is_leaf_alloc(var)) this->Free(var); + this->Free(var); } } // scope_pair_offset >= 0 means it is either @@ -494,11 +490,6 @@ class SharedMemoryRewriter : public StmtExprMutator { alloc_map_[var] = dst_entry; } } - if (it != event_map_.end() && seq[i].scope_pair_offset <= 0) { - for (const VarNode* var : it->second.kill) { - if (is_leaf_alloc(var)) this->Free(var); - } - } } } /*! @@ -545,7 +536,6 @@ class SharedMemoryRewriter : public StmtExprMutator { StorageEntry* e = it->second; e->const_nbits = std::max(const_nbits, e->const_nbits); const_free_map_.erase(it); - it->second->allocs.push_back({op->buffer_var.get()}); return e; } // Then start looking at smaller buffers.
