hanhanW wrote: https://github.com/llvm/torch-mlir/pull/4370 has the fix, which works for IREE. There are torch-mlir failure in CI, and maybe @sjarus you can pick it up from here. https://github.com/llvm/llvm-project/pull/166904