add placeholder to inject_vpthread

apache · Feb 10, 2021 · 896b02f · 896b02f
1 parent 00c17d9
commit 896b02f
Showing 1 changed file with 6 additions and 0 deletions.
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);