gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[FRONTEND] Fixed python bindings link options (#40)

Browse Source
This commit is contained in:
Philippe Tillet
2022-08-07 13:09:12 -07:00
committed by GitHub
parent 78ebbe24c7
commit 490d34e0d5
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
CMakeLists.txt
View File

@@ -176,15 +176,17 @@ add_library(triton SHARED ${PYTHON_SRC})
set(TRITON_SOURCE_DIR "${CMAKE_CURRENT_SOURCE_DIR}") set(TRITON_SOURCE_DIR "${CMAKE_CURRENT_SOURCE_DIR}")
set(TRITON_BINARY_DIR "${CMAKE_CURRENT_BINARY_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 target_link_libraries(triton
${PYTHON_LIBRARIES} ${PYTHON_LIBRARIES}
TritonIR
TritonAnalysis TritonAnalysis
TritonTransforms TritonTransforms
TritonDriver
TritonToTritonGPU
TritonGPUIR
TritonGPUTransforms TritonGPUTransforms
TritonDriver
${dialect_libs}
${conversion_libs}
# optimizations # optimizations
MLIRPass MLIRPass
MLIRTransforms MLIRTransforms