diff --git a/lib/Conversion/TorchConversionToMLProgram/CMakeLists.txt b/lib/Conversion/TorchConversionToMLProgram/CMakeLists.txt
index f819ad018..b89ffbb43 100644
--- a/lib/Conversion/TorchConversionToMLProgram/CMakeLists.txt
+++ b/lib/Conversion/TorchConversionToMLProgram/CMakeLists.txt
@@ -12,9 +12,10 @@ add_mlir_conversion_library(TorchMLIRTorchConversionToMLProgram
 
   LINK_LIBS PUBLIC
   MLIRIR
-  MLIRPass
   MLIRLinalgDialect
+  MLIRMLProgramDialect
   MLIRMathDialect
+  MLIRPass
   TorchMLIRTorchDialect
 )
 
diff --git a/lib/Conversion/TorchToTMTensor/CMakeLists.txt b/lib/Conversion/TorchToTMTensor/CMakeLists.txt
index eda9ec7ad..d05d8277c 100644
--- a/lib/Conversion/TorchToTMTensor/CMakeLists.txt
+++ b/lib/Conversion/TorchToTMTensor/CMakeLists.txt
@@ -17,6 +17,7 @@ TorchToTMTensor.cpp
   MLIRMathDialect
   TorchMLIRTorchDialect
   TorchMLIRTMTensorDialect
+  TorchMLIRTorchUtils
 )
 
 torch_mlir_target_includes(TorchMLIRTorchToTMTensor)
diff --git a/lib/Conversion/TorchToTosa/CMakeLists.txt b/lib/Conversion/TorchToTosa/CMakeLists.txt
index e1f5142bd..909ee3bcb 100644
--- a/lib/Conversion/TorchToTosa/CMakeLists.txt
+++ b/lib/Conversion/TorchToTosa/CMakeLists.txt
@@ -16,6 +16,7 @@ add_mlir_conversion_library(TorchMLIRTorchToTosa
   MLIRIR
   MLIRPass
   MLIRTosaDialect
+  TorchMLIRConversionUtils
   TorchMLIRTorchDialect
 )
 
diff --git a/lib/Dialect/TorchConversion/Transforms/CMakeLists.txt b/lib/Dialect/TorchConversion/Transforms/CMakeLists.txt
index a5d5f9b70..1f7f4e8f8 100644
--- a/lib/Dialect/TorchConversion/Transforms/CMakeLists.txt
+++ b/lib/Dialect/TorchConversion/Transforms/CMakeLists.txt
@@ -1,16 +1,22 @@
-set(LinkedLibs  MLIRIR
-                MLIRPass
-                MLIRFuncTransforms
-                TorchMLIRTorchConversionDialect
-                TorchMLIRTorchDialect
-                TorchMLIRTorchPasses
-                TorchMLIRTorchToLinalg
-                TorchMLIRTorchToTMTensor
-                TorchMLIRTorchToArith
-                TorchMLIRTorchToSCF
-                TorchMLIRTorchConversionToMLProgram
-                MLIRMemRefTransforms)
-                
+set(LinkedLibs
+  MLIRFuncTransforms
+  MLIRIR
+  MLIRLinalgTransforms
+  MLIRMemRefTransforms
+  MLIRPass
+  MLIRTosaTransforms
+  MLIRVectorTransforms
+  TorchMLIRTorchConversionDialect
+  TorchMLIRTorchConversionToMLProgram
+  TorchMLIRTorchDialect
+  TorchMLIRTorchPasses
+  TorchMLIRTorchToArith
+  TorchMLIRTorchToLinalg
+  TorchMLIRTorchToSCF
+  TorchMLIRTorchToTMTensor
+  TorchMLIRTorchToTosa
+  )
+
 if(TORCH_MLIR_ENABLE_STABLEHLO)
   list(APPEND LinkedLibs ChloPasses)
 endif()