Skip to content

Commit 2b24609

Browse files
authored
Link triton-shared with pybind (#316)
Required to make triton-shared work with pybind
1 parent e15146b commit 2b24609

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

CMakeLists.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,4 +15,5 @@ add_subdirectory(tools/triton-shared-opt)
1515

1616
if (TRITON_SHARED_BUILD_CPU_BACKEND)
1717
add_triton_plugin(TritonShared ${CMAKE_CURRENT_SOURCE_DIR}/triton_shared.cc LINK_LIBS TritonSharedAnalysis TritonToLinalg TritonTilingExtIR)
18+
target_link_libraries(TritonShared PRIVATE Python3::Module pybind11::headers ${Python3_LIBRARIES})
1819
endif()

0 commit comments

Comments
 (0)