Thank you @jroesch, that definitely brings in more clarity.
The way I was using this pass in its previous version was to call it on stmt within my own pass as a step after some IR transformations, and it worked flawlessly. Looking at your answer, it seems that what I am looking for is the Python API equivalent of > arith::Analyzer analyzer; > arith::StmtSimplifier(&analyzer).Simplify(stmt) Basically what I was looking for is being able to run the Simplify pass in my own pass at a function or statement granularity rather than the whole module. --- [Visit Topic](https://discuss.tvm.apache.org/t/how-to-leverage-tvm-tir-transform-in-tvm-0-7/7780/4) to respond. You are receiving this because you enabled mailing list mode. To unsubscribe from these emails, [click here](https://discuss.tvm.apache.org/email/unsubscribe/14d178183a2c6a346bbdf609f874b5a733a02396ea129a0d2c0a3e5e1d9c9d2a).