Couldn’t this be implemented as a custom IR pass (in Python or C++) instead of as a new scheduling primitive? This is essentially taking the body `b` of a `For` and replacing it with `Block(prologue, b, epilogue)` right?
--- [Visit Topic](https://discuss.tvm.ai/t/add-support-for-extern-prologue-epilogue-functions/6041/5) to respond. You are receiving this because you enabled mailing list mode. To unsubscribe from these emails, [click here](https://discuss.tvm.ai/email/unsubscribe/173f62903e5ea7d2490f4c43c0912941a55ecf44e898461905cbaa010b79dbd5).