How do I add custom data load and write back through TIR?

Thank you for your answer. I would like to know how to tag read/write stages with specific pragmas.