https://github.com/jurahul approved this pull request. Changes LGTM, but let's wait for another approval from someone more active on MLIR side as well (@jpienaar or @River707 ?) https://github.com/llvm/llvm-project/pull/126714