The way to do it is using the "pragma" method of the stage. There are examples [here](https://github.com/apache/tvm/blob/6fca5c657a2fadc16fd7ff44de8a6a9656d50c1b/vta/python/vta/top/vta_conv2d.py#L148) and [here](https://github.com/apache/tvm/blob/6fca5c657a2fadc16fd7ff44de8a6a9656d50c1b/apps/topi_recipe/gemm/gemm_int8.py#L119).
--- [Visit Topic](https://discuss.tvm.apache.org/t/how-do-i-add-custom-data-load-and-write-back-through-tir/12927/4) 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/86009dadbc16dc42bb6baeaafc29111382cdfd080b746a2a904fe8ca805f0546).
