From 07df1a08725866f7fb309ae905fbc53e312df030 Mon Sep 17 00:00:00 2001
From: wrongtest <wrongtest0@gmail.com>
Date: Sun, 21 Aug 2022 04:55:35 +0800
Subject: [PATCH] fix temp array object reference in
 manifest_shared_memory_local_stage (#12516)

---
 src/tir/transforms/manifest_shared_memory_local_stage.cc | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/tir/transforms/manifest_shared_memory_local_stage.cc b/src/tir/transforms/manifest_shared_memory_local_stage.cc
index 3a3abf0b801c..16c85642d1e5 100644
--- a/src/tir/transforms/manifest_shared_memory_local_stage.cc
+++ b/src/tir/transforms/manifest_shared_memory_local_stage.cc
@@ -46,7 +46,7 @@ namespace tir {
  */
 class IntermediateStageRewriter {
  public:
-  explicit IntermediateStageRewriter(const Array<Stmt>& ancestor_loop_or_blocks)
+  explicit IntermediateStageRewriter(const std::vector<Stmt>& ancestor_loop_or_blocks)
       : ancestor_loop_or_blocks_(ancestor_loop_or_blocks) {}
 
   std::tuple<Buffer, Buffer, Block, Stmt> Rewrite(const BlockNode* block) {
@@ -165,7 +165,7 @@ class IntermediateStageRewriter {
     return {new_buffer, buffer_indices};
   }
 
-  const Array<Stmt>& ancestor_loop_or_blocks_;
+  const std::vector<Stmt>& ancestor_loop_or_blocks_;
 };
 
 class SharedMemoryLocalStageInserter : public StmtMutator {