Skip to content

Commit eb31123

Browse files
fix temp array object reference in manifest_shared_memory_local_stage (#12516)
1 parent 1985c01 commit eb31123

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/tir/transforms/manifest_shared_memory_local_stage.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ namespace tir {
4646
*/
4747
class IntermediateStageRewriter {
4848
public:
49-
explicit IntermediateStageRewriter(const Array<Stmt>& ancestor_loop_or_blocks)
49+
explicit IntermediateStageRewriter(const std::vector<Stmt>& ancestor_loop_or_blocks)
5050
: ancestor_loop_or_blocks_(ancestor_loop_or_blocks) {}
5151

5252
std::tuple<Buffer, Buffer, Block, Stmt> Rewrite(const BlockNode* block) {
@@ -165,7 +165,7 @@ class IntermediateStageRewriter {
165165
return {new_buffer, buffer_indices};
166166
}
167167

168-
const Array<Stmt>& ancestor_loop_or_blocks_;
168+
const std::vector<Stmt>& ancestor_loop_or_blocks_;
169169
};
170170

171171
class SharedMemoryLocalStageInserter : public StmtMutator {

0 commit comments

Comments
 (0)