[Apache TVM Discuss] [Development] Creating store_at in TVM

[Matt Barrett via Apache TVM Discuss <no...@discuss.tvm.ai>]

Halide provides the scheduling primitive `store_at` (as well as `store_root`) to move where the storage of a tensor happens independent of the compute. This is very useful for when we want to make use of the sliding window optimization and create rolling buffers - both of which can be critical in reducing memory usage on memory constrained devices. For examples of how this is used in Halide, you can reference the tutorial on multi-stage pipelines: https://halide-lang.org/tutorials/tutorial_lesson_08_scheduling_2.html.

Is this something we can emulate in TVM with the existing scheduling intrinsics? And if not, is this something the design of TE would permit? In the latter case, I'd be interested to know whether it would currently be worth implementing in TE given the change of approach in TensorIR, or whether it would be better to wait.

Thanks

cc @junrushao1994 @tqchen  @manupa-arm





---
[Visit Topic](https://discuss.tvm.apache.org/t/creating-store-at-in-tvm/8083/1) 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/b2f67173b676bcd17bdbfae4fbc081207956e35c0740a6e5cebdd664bd29a2a3).

[Apache TVM Discuss] [Development] Creating store_at in TVM

[Manupa Karunaratne via Apache TVM Discuss <no...@discuss.tvm.ai>]

Yes definitely useful to have! might save a lot of hacks/workarounds that would otherwise needed to get the same functionality. Also cc : @spectrometerHBH @merrymercy





---
[Visit Topic](https://discuss.tvm.apache.org/t/creating-store-at-in-tvm/8083/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/6cae5932ea4e8f2728287de6973573b60848301b4819919a3a8e3366d875b1d9).