[WIP][LinalgExt] Support iree_linalg_ext.index in attention ops for causal masks#24056

keshavvinayak01 wants to merge 1 commit intoiree-org:mainiree-org/iree:mainfrom

keshavvinayak01:users/keshavvinayak01/causal-index-fusekeshavvinayak01/iree:users/keshavvinayak01/causal-index-fuseCopy head branch name to clipboard