[mlir] Add missing build deps for Mesh transforms

[kuhar]

No description provided.

[sogartar]

I think the CMake also requires these dependencies.

[sogartar]

Lets unbloc the Bazel build quickly. I will open a PR to fix the CMake dependencies as well.

[kuhar]

I think the CMake also requires these dependencies.

Added

[llvmbot]

@llvm/pr-subscribers-mlir

Author: Jakub Kuderski (kuhar)

Changes

---

Full diff: https://github.com/llvm/llvm-project/pull/84581.diff

2 Files Affected:

• (modified) mlir/lib/Dialect/Mesh/Transforms/CMakeLists.txt (+2)
• (modified) utils/bazel/llvm-project-overlay/mlir/BUILD.bazel (+2)

diff --git a/mlir/lib/Dialect/Mesh/Transforms/CMakeLists.txt b/mlir/lib/Dialect/Mesh/Transforms/CMakeLists.txt
index 28af820440076c..212ea6d6948b27 100644
--- a/mlir/lib/Dialect/Mesh/Transforms/CMakeLists.txt
+++ b/mlir/lib/Dialect/Mesh/Transforms/CMakeLists.txt
@@ -17,7 +17,9 @@ add_mlir_dialect_library(MLIRMeshTransforms
   MLIRArithDialect
   MLIRArithUtils
   MLIRControlFlowDialect
+  MLIRControlFlowInterfaces
   MLIRFuncDialect
+  MLIRFunctionInterfaces
   MLIRIR
   MLIRMeshDialect
   MLIRPass
diff --git a/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel b/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel
index cadcbcc5a0922b..3be8f34ae0f4d5 100644
--- a/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel
+++ b/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel
@@ -3410,8 +3410,10 @@ cc_library(
         ":ArithDialect",
         ":ArithUtils",
         ":ControlFlowDialect",
+        ":ControlFlowInterfaces",
         ":DialectUtils",
         ":FuncDialect",
+        ":FunctionInterfaces",
         ":IR",
         ":MeshDialect",
         ":MeshShardingInterface",