(tvm) 06/06: fix typo in storage rewrite

[sy...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

syfeng pushed a commit to branch test_all_cases_on_unity
in repository https://gitbox.apache.org/repos/asf/tvm.git

commit 82fe0b25567b1bb2265123a61c1db81a14fdc8ea
Author: Siyuan Feng <hz...@sjtu.edu.cn>
AuthorDate: Sun Jan 21 22:56:37 2024 +0800

    fix typo in storage rewrite
---
 src/tir/transforms/storage_rewrite.cc | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/tir/transforms/storage_rewrite.cc b/src/tir/transforms/storage_rewrite.cc
index 7f48a53708..6875523a95 100644
--- a/src/tir/transforms/storage_rewrite.cc
+++ b/src/tir/transforms/storage_rewrite.cc
@@ -1710,8 +1710,11 @@ Pass StorageRewrite() {
     if (target.defined() && target->kind->name == "vulkan") {
       merge_static_smem = false;
     }
+    // Only enable reuse when we are not merging static shared memory.
+    // Otherwise we will do it in a separate stage
+    bool enable_reuse = merge_static_smem ? false : true;
     auto* n = f.CopyOnWrite();
-    n->body = StoragePlanRewriter().Rewrite(std::move(n->body), true, merge_static_smem);
+    n->body = StoragePlanRewriter().Rewrite(std::move(n->body), true, enable_reuse);
     // Parameters may not be rewritten, but internal allocations may.
     // Vectorization of AllocateConst is currently disabled, as it has
     // indexing issues for types that include padding (e.g. int8x3