[Texture] Add 2d memory support into static memory planner#11876
[Texture] Add 2d memory support into static memory planner#11876masahi merged 2 commits intoapache:mainfrom
Conversation
|
@csullivan @masahi @mbs-octoml Please review and push this PR through. |
|
@elvin-n Can you add tests? |
There was a problem hiding this comment.
These changes to the graph memory planner originate back from #7690. Given that I was the author of that PR and many of these changes, I don't feel comfortable merging this without additional reviews but the change looks good to me if others feel that refactor the graph memory planner in this way is acceptable.
Ultimately in Relax graph-tir memory planning will be unified and texture can be represented as an Nd Buffer, and a separate pass can handle memory planning based on a target specific handling for Nd memory. Given that I'm personally okay with this change landing as even though it is fairly specific, we don't intend for the graph executor memory planner to be any more generically useful for Nd memory down the road.
Edit: As @masahi pointed out, it would be useful to have some unit tests of the 2d planner.
Co-authored-by: Chris Sullivan <csullivan@octoml.ai>
3cf99ab to
9eab9b2
Compare
Added test |
Co-authored-by: Chris Sullivan csullivan@octoml.ai
This PR is a split part of origin PR11357
@csullivan @mbs-octoml