https://github.com/kuhar commented: Thanks for the PR. Would you be able to add some example use case that we could add to the PR description? IIRC this is intended for flash attention? https://github.com/llvm/llvm-project/pull/104851