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

[PYTHON] Some cleaning of the PyBind11 wrappers (#62)

Browse Source
This commit is contained in:
Philippe Tillet
2021-02-06 17:10:44 -08:00
committed by Philippe Tillet
parent 08909b49c8
commit 2a02fabdac
8 changed files with 282 additions and 350 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
CMakeLists.txt
View File

@@ -33,8 +33,8 @@ endif()
if(BUILD_PYTHON_MODULE)
message(STATUS "Adding Python module")
# PyBind11 wrapper source file
set(TORCH_SRC torch/launch.cc torch/superblock.cc)
set(PYTHON_SRC bindings.cc ${TORCH_SRC})
file(GLOB_RECURSE TORCH_SRC torch/*.cc)
set(PYTHON_SRC main.cc triton.cc ${TORCH_SRC})
set_source_files_properties(${TORCH_SRC} PROPERTIES COMPILE_FLAGS "-std=c++14 -D_GLIBCXX_USE_CXX11_ABI=${TORCH_CXX11_ABI}")
include_directories("." ${PYTHON_INCLUDE_DIRS})
link_directories(${PYTHON_LINK_DIRS})