Add lift_if_then_else pass (#3865)

* Add LiftIfThenElse pass * Add more comments * Rename and refactor * Add description for internal data structure * Rename a test * Minor change * Address comments * Improve update_for
apache · Oct 18, 2019 · 687d4a8 · 687d4a8
1 parent fdb01cb
commit 687d4a8
Show file tree

Hide file tree

Showing 4 changed files with 617 additions and 0 deletions.
diff --git a/include/tvm/ir_pass.h b/include/tvm/ir_pass.h
@@ -377,6 +377,13 @@ Stmt LowerStorageAccessInfo(Stmt stmt);
  */
 Stmt DecorateDeviceScope(Stmt stmt);
 
+/*!
+ * \brief Loop invariant code motion which locates and hoists if statements.
+ * \param stmt The stmt to do if statement hoisting.
+ * \return Transformed stmt.
+ */
+Stmt HoistIfThenElse(Stmt stmt);
+
 /*!
  * \brief Make an user callable API LoweredFunc.
  *

diff --git a/src/api/api_pass.cc b/src/api/api_pass.cc
@@ -160,5 +160,6 @@ REGISTER_PASS(VerifyGPUCode);
 REGISTER_PASS(DecorateDeviceScope);
 REGISTER_PASS(InstrumentBoundCheckers);
 REGISTER_PASS(VerifyCompactBuffer);
+REGISTER_PASS(HoistIfThenElse);
 }  // namespace ir
 }  // namespace tvm