Skip to content

Commit 40970f4

Browse files
committed
Link triton-shared with pybind
1 parent 859ede9 commit 40970f4

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

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)