I'm no expert with this, but if you take a look in build_module.py, you can see what TIR passes are run after the schedules have been lowered. I'll paste them here for convenience: ``` tvm.tir.transform.InjectPrefetch(), tvm.tir.transform.StorageFlatten(64, instrument_bound_checkers), tvm.tir.transform.BF16Legalize(), tvm.tir.transform.NarrowDataType(32), tvm.tir.transform.Simplify(), tvm.tir.transform.LoopPartition(), tvm.tir.transform.VectorizeLoop(not disable_vectorize), tvm.tir.transform.InjectVirtualThread(), tvm.tir.transform.InjectDoubleBuffer(), tvm.tir.transform.StorageRewrite(), tvm.tir.transform.UnrollLoop(), tvm.tir.transform.Simplify(), tvm.tir.transform.RemoveNoOp(), tvm.tir.transform.RewriteUnsafeSelect(), tvm.tir.transform.HoistIfThenElse() ``` I'm only familiar with a few of these, but sometimes to get a better understanding I'll just print the module between passes and see what changed.
--- [Visit Topic](https://discuss.tvm.apache.org/t/tir-after-auto-tuning-what-other-optimizations-based-on-tir-will-be-done/8360/2) 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/5342443c671e257ab1fd63c822fa233da1929e48bc5346a97d4305c0f2a02a90).