tvm icon indicating copy to clipboard operation
tvm copied to clipboard

[TIR] Output DeclBuffer in FlattenBuffer

Open Lunderberg opened this issue 2 years ago • 1 comments

If a flattened buffer is produced for use in BufferLoad and BufferStore statements, generate a DeclBuffer. This PR updates the behavior for both TE-derived schedules (tir.StorageFlatten transform) and TIR-derived schedules (tir.FlattenBuffer transform).

This is a subset of the changes made in https://github.com/apache/tvm/pull/14778, broken out for ease of testing and review.

Lunderberg avatar Sep 15 '23 13:09 Lunderberg

This is blocked on the ethos-u and ethos-m codegen, which does not currently support DeclBuffer nodes.

Lunderberg avatar Sep 12 '24 18:09 Lunderberg