From 490d34e0d5e6e0b7671b8e1f9fcb1c6b9db84846 Mon Sep 17 00:00:00 2001
From: Philippe Tillet <phil@openai.com>
Date: Sun, 7 Aug 2022 13:09:12 -0700
Subject: [PATCH] [FRONTEND] Fixed python bindings link options (#40)

---
 CMakeLists.txt | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

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