fix temp array object reference in manifest_shared_memory_local_stage (…

…apache#12516)
daobook · Nov 25, 2022 · 07df1a0 · 07df1a0
1 parent ea69ebf
commit 07df1a0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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 {