From 3b5a0b74c31e064d4dcb333f8e8c550a98150209 Mon Sep 17 00:00:00 2001
From: Jacques Pienaar <jpienaar@google.com>
Date: Thu, 31 Aug 2023 11:31:36 -0700
Subject: [PATCH] Update cmake files

---
 .../compiler/InputConversion/StableHLO/Preprocessing/BUILD.bazel | 1 +
 .../InputConversion/StableHLO/Preprocessing/CMakeLists.txt       | 1 +
 2 files changed, 2 insertions(+)

diff --git a/compiler/src/iree/compiler/InputConversion/StableHLO/Preprocessing/BUILD.bazel b/compiler/src/iree/compiler/InputConversion/StableHLO/Preprocessing/BUILD.bazel
index a4f6958058832..bf743ddfc85c2 100644
--- 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
index d76bb7ed8b0fb..31169c704eac1 100644
--- 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