I would advise you to take a look at the cache_read and cache_write methods of the schedules. In my particular case (I don't know if this is exactly what you are looking for) I use this 2 methods to generate read and write stages, and then tag this stages with specific pragmas. Then, I have 2 separate passes that replace the loops tagged with this pragmas with my own load and store macros.
--- [Visit Topic](https://discuss.tvm.apache.org/t/how-do-i-add-custom-data-load-and-write-back-through-tir/12927/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/211df137ec43395e4665b5970e131d76c72ed4373d1a68c98722cb53369db27b).
