diff --git a/lib/CAPI/CMakeLists.txt b/lib/CAPI/CMakeLists.txt
index 1fe50de93..173ff2878 100644
--- 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
index 7d4555c90..5abb141f4 100644
--- 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
index 3d839bffc..feeb55015 100644
--- 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