https://github.com/MaheshRavishankar requested changes to this pull request. Dismissed prematurely. Just want to get clarification on the region of the op, but otherwise this looks good. https://github.com/llvm/llvm-project/pull/123618