Skip to content

[TMTensor] Remove Materialization of causal_masks; add attribute instead#4520

Draft
keshavvinayak01 wants to merge 2 commits intollvm:mainfrom
keshavvinayak01:users/keshavvinayak01/TMTensor-causal-mask-attr
Draft

[TMTensor] Remove Materialization of causal_masks; add attribute instead#4520
keshavvinayak01 wants to merge 2 commits intollvm:mainfrom
keshavvinayak01:users/keshavvinayak01/TMTensor-causal-mask-attr

Conversation

@keshavvinayak01
Copy link
Copy Markdown
Contributor

@keshavvinayak01 keshavvinayak01 commented Apr 2, 2026

Removing the materialisation of simple causal masks removes unnecessary buffer allocations and enables fusion opportunities downstream. (Example: iree-org/iree#23999)

Signed-off-by: Keshav Vinayak Jha <keshavvinayakjha@gmail.com>
Signed-off-by: Keshav Vinayak Jha <keshavvinayakjha@gmail.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant