We have moved to having all passes use the pass infrastructure now, you should be able to just invoke Simplify on your entire IRModule before running your pass. i.e Simplify()(mod).
We have moved to having all passes use the pass infrastructure now, you should be able to just invoke Simplify on your entire IRModule before running your pass. i.e Simplify()(mod).