New python binding

2022-03-22 21:53:22 +08:00
parent 419bbe0f6e
commit f2ab318614
5 changed files with 593 additions and 461 deletions
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -174,10 +174,15 @@ add_subdirectory(lib)

 add_library(triton SHARED ${PYTHON_SRC})

+find_package(PythonLibs REQUIRED)
+
 target_link_libraries(triton
  TritonIR
  TritonDriver
-  TritonCodeGen
+  # TritonCodeGen
+
+  MLIRCAPIIR
+  ${PYTHON_LIBRARIES}
 )

 target_link_options(triton PRIVATE ${LLVM_LDFLAGS})