diff --git a/CMakeLists.txt b/CMakeLists.txt
index 7971d3aed..c8ec261f6 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -176,15 +176,17 @@ add_library(triton SHARED ${PYTHON_SRC})
 set(TRITON_SOURCE_DIR "${CMAKE_CURRENT_SOURCE_DIR}")
 set(TRITON_BINARY_DIR "${CMAKE_CURRENT_BINARY_DIR}")
 
+get_property(dialect_libs GLOBAL PROPERTY MLIR_DIALECT_LIBS)
+get_property(conversion_libs GLOBAL PROPERTY MLIR_CONVERSION_LIBS)
+
 target_link_libraries(triton
   ${PYTHON_LIBRARIES}
-  TritonIR
   TritonAnalysis
   TritonTransforms
-  TritonDriver
-  TritonToTritonGPU
-  TritonGPUIR
   TritonGPUTransforms
+  TritonDriver
+  ${dialect_libs}
+  ${conversion_libs}
   # optimizations
   MLIRPass
   MLIRTransforms