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 <[email protected]> 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
