[TORCH][MLIR] Added _sdpa_flash_attention op#4417

keshavvinayak01 wants to merge 14 commits intollvm:mainllvm/torch-mlir:mainfrom

keshavvinayak01:keshavvinayak01/torch-aten-flashattention-sdpa-rewritekeshavvinayak01/torch-mlir:keshavvinayak01/torch-aten-flashattention-sdpa-rewriteCopy head branch name to clipboard