Fix missing CMake deps for THLO

Fixes the immediate problem behind #45 using patches by Stella. This CL doesn't aim to address the layering part of the issue, e.g. MHLO passes depending on the THLO dialect. For now, it just fixed the blocker on the IREE side where a symbol in a shared library (the THLO type descriptor) is failing to resolve at runtime. PiperOrigin-RevId: 467966552
tensorflow · Aug 16, 2022 · 3771b40 · 3771b40
1 parent 282ffc2
commit 3771b40
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git a/lib/CAPI/CMakeLists.txt b/lib/CAPI/CMakeLists.txt
@@ -6,6 +6,7 @@ add_mlir_public_c_api_library(MLIRHLOCAPIDialects
   LINK_LIBS PUBLIC
   ChloDialect
   MhloDialect
+  THLODialect
   # For AllMhLoPasses:
   ChloPasses
   MhloPasses

diff --git a/lib/Dialect/mhlo/transforms/CMakeLists.txt b/lib/Dialect/mhlo/transforms/CMakeLists.txt
@@ -78,6 +78,7 @@ add_mlir_library(MhloPasses
   MLIRMhloUtils
   MLIRPass
   MLIRTransformUtils
+  THLODialect
 )
 
 add_mlir_library(MhloToLhloConversion

diff --git a/lib/Dialect/thlo/IR/CMakeLists.txt b/lib/Dialect/thlo/IR/CMakeLists.txt
@@ -25,6 +25,7 @@ add_mlir_dialect_library(THLODialect
   MLIRGmlStFusionInterfaceIncGen
 
   LINK_LIBS PUBLIC
+  GmlStDialect
   MLIRIR
   MLIRMemRefDialect
   MLIRSideEffectInterfaces