[mlir] Add support for broader range of input files in generate-test-checks.py#134327

GleasonK merged 2 commits intollvm:mainllvm/llvm-project:mainfrom

GleasonK:gen-test-checksGleasonK/llvm-project:gen-test-checksCopy head branch name to clipboard