Update cmake files

iree-org · Aug 31, 2023 · 3b5a0b7 · 3b5a0b7
1 parent 67bbf9c
commit 3b5a0b7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/compiler/src/iree/compiler/InputConversion/StableHLO/Preprocessing/BUILD.bazel b/compiler/src/iree/compiler/InputConversion/StableHLO/Preprocessing/BUILD.bazel
@@ -86,6 +86,7 @@ iree_compiler_cc_library(
         "@llvm-project//mlir:FuncDialect",
         "@llvm-project//mlir:IR",
         "@llvm-project//mlir:MathDialect",
+        "@llvm-project//mlir:MLProgramDialect",
         "@llvm-project//mlir:Pass",
         "@llvm-project//mlir:SCFDialect",
         "@llvm-project//mlir:ShapeDialect",

diff --git a/compiler/src/iree/compiler/InputConversion/StableHLO/Preprocessing/CMakeLists.txt b/compiler/src/iree/compiler/InputConversion/StableHLO/Preprocessing/CMakeLists.txt
@@ -74,6 +74,7 @@ iree_cc_library(
     MLIRFuncDialect
     MLIRIR
     MLIRMathDialect
+    MLIRMLProgramDialect
     MLIRPass
     MLIRSCFDialect
     MLIRShapeDialect