[llvm-branch-commits] [mlir] Add missing imports (PR #138550)

[Mehdi Amini via llvm-branch-commits]

joker-eph wrote:

In general, I believe we cherry-pick fixes commits from `main` instead of rewriting patches directly to the release branch.

https://github.com/llvm/llvm-project/pull/138550