Thanks a lot for the detailed explanation @tqchen and @slyubomirsky
I understand the concern about modifying the StructInfo. My only thought was to store the logical to physical memory mapping somewhere in Relax, so that when we do static memory planning, we don't just assume 1d allocation, but compute the actual physical memory shape. StructInfo just seemed like a logical place to do this as that is used by the current memory planning/codegen pass to calculate the memory required for a tensor, but we can discuss other alternatives. I'll wait for @abhikran-quic to also reply to get his perspective, but yes I agree that it might be great to talk it out in a dev meeting to understand the perspectives and come up with some ideas. --- [Visit Topic](https://discuss.tvm.apache.org/t/axis-separator-in-relax-tensor/15385/6) 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/a1fa788d8eb913dd18496510b95507718abf2618a95f6ebeb26f0301f47e6103).