|
10 | 10 | #include <tvm/tir/stmt_functor.h> |
11 | 11 | #include <tvm/tir/transform.h> |
12 | 12 |
|
13 | | -#include <utility> |
14 | 13 | #include <unordered_set> |
| 14 | +#include <utility> |
15 | 15 |
|
16 | 16 | #include "../op/builtin.h" |
17 | 17 |
|
@@ -287,8 +287,7 @@ class MultiVersionBufferRewriter : public StmtExprMutator { |
287 | 287 | LOG(FATAL) |
288 | 288 | << "MultiVersionBuffer: Can't handle the body of the loop because " |
289 | 289 | << "it is not a SeqStmt, IfThenElse without else, " |
290 | | - << "or LetStmt wrapping them, but got " |
291 | | - << current->GetTypeKey(); |
| 290 | + << "or LetStmt wrapping them, but got " << current->GetTypeKey(); |
292 | 291 | } |
293 | 292 | } |
294 | 293 | ICHECK(pipeline_body_seq != nullptr); |
@@ -422,12 +421,14 @@ class MultiVersionBufferRewriter : public StmtExprMutator { |
422 | 421 |
|
423 | 422 | PrimExpr version_index_; |
424 | 423 | std::vector<std::pair<Var, PrimExpr>> loop_stack_; |
425 | | - // Track ancestor statements to query whether an LCA is inside the current loop. |
| 424 | + // Track ancestor statements to query whether an LCA is inside the current |
| 425 | + // loop. |
426 | 426 | std::vector<const StmtNode *> stmt_stack_; |
427 | 427 | Map<Var, Buffer> buffer_data_to_buffer_; |
428 | 428 | Map<Buffer, Optional<Stmt>> buffer_lca_; |
429 | 429 | Map<Buffer, Buffer> buffer_remap_; |
430 | | - // Remember each block's alloc list so the loop can see buffers defined in parents. |
| 430 | + // Remember each block's alloc list so the loop can see buffers defined in |
| 431 | + // parents. |
431 | 432 | std::unordered_map<const BlockNode *, Array<Buffer>> block_alloc_buffers_; |
432 | 433 | }; |
433 | 434 |
|
|
0 commit comments