add placeholder to inject_vpthread

masahi · masahi · commit 896b02fb8aba · 2021-02-10T22:04:56.000+09:00
diff --git a/src/tir/transforms/inject_virtual_thread.cc b/src/tir/transforms/inject_virtual_thread.cc
@@ -333,6 +333,12 @@ class VTInjector : public StmtExprMutator {
     }
   }
 
+  // While
+  Stmt VisitStmt_(const WhileNode* op) final {
+    // TODO(masahi): Do we need a special handling for While nodes?
+    return StmtMutator::VisitStmt_(op);
+  }
+
   // Seq
   Stmt VisitStmt_(const SeqStmtNode* op) final {
     ICHECK_EQ(max_loop_depth_, 0);

Original file line number	Diff line number	Diff line change
`@@ -333,6 +333,12 @@ class VTInjector : public StmtExprMutator {`
`333`	`333`	`}`
`334`	`334`	`}`
`335`	`335`
	`336`	`+ // While`
	`337`	`+ Stmt VisitStmt_(const WhileNode* op) final {`
	`338`	`+ // TODO(masahi): Do we need a special handling for While nodes?`
	`339`	`+ return StmtMutator::VisitStmt_(op);`
	`340`	`+ }`
	`341`	`+`
`336`	`342`	`// Seq`
`337`	`343`	`Stmt VisitStmt_(const SeqStmtNode* op) final {`
`338`	`344`	`ICHECK_EQ(max_loop_depth_, 0);`