diff --git a/src/tir/transforms/plan_update_buffer_allocation_location.cc b/src/tir/transforms/plan_update_buffer_allocation_location.cc index 4701d86092b50..59f9170786b6b 100644 --- a/src/tir/transforms/plan_update_buffer_allocation_location.cc +++ b/src/tir/transforms/plan_update_buffer_allocation_location.cc @@ -93,7 +93,8 @@ class BufferAllocationLocator : public StmtExprMutator { op = stmt.as(); ICHECK(op != nullptr); - // No longer consider buffers created by match_buffer inside the block when updating access region. + // No longer consider buffers created by match_buffer inside the block when updating access + // region. for (const MatchBufferRegion match_buffer : op->match_buffers) { const Var& target_var = match_buffer->buffer->data; buffer_data_to_buffer_.erase(target_var); @@ -108,8 +109,8 @@ class BufferAllocationLocator : public StmtExprMutator { ObjectPtr n = CopyOnWrite(op); n->alloc_buffers = std::move(alloc_buffers); // Erase buffer allocated inside the block from access region. - n->reads = UpdateRegion(n->reads); - n->writes = UpdateRegion(n->writes); + n->reads = RemoveRedundantBufferRegion(n->reads); + n->writes = RemoveRedundantBufferRegion(n->writes); return Stmt(n); } @@ -133,7 +134,7 @@ class BufferAllocationLocator : public StmtExprMutator { return std::move(realize); } - Array UpdateRegion(const Array& region) const { + Array RemoveRedundantBufferRegion(const Array& region) const { Array result; for (const BufferRegion& buffer_region : region) { if (buffer_data_to_buffer_.count(buffer_region->buffer->data)) {