[BACKEND] Use the LL API to replace the using of legacy layout attribute API.

triton-lang / triton

Development repository for the Triton language and compiler

https://triton-lang.org/

MIT License

13.47k stars 1.66k forks source link

[BACKEND] Use the LL API to replace the using of legacy layout attribute API. #5196

Closed chengjunlu closed 1 day ago

chengjunlu commented 2 days ago

The util function getDistributedLayoutStr uses the DistributedLayout attribute interface, which is not flexible for third-party extensions. Use the getInDimSize of the LinearLayout, which is better since the legacy layout has been converted to the LinearLayout.

There is no new test case since it is only a change in API usage.