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

[BUILD] Add TRITON Prefix to build variables (#752)

Browse Source
This commit is contained in:
Ian Bearman
2022-10-09 10:55:17 -07:00
committed by GitHub
parent 1d772cd843
commit 448d14a598
4 changed files with 11 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
.gitignore vendored
View File

@@ -7,3 +7,6 @@ python/build/
python/triton.egg-info/ python/triton.egg-info/
python/triton/_C/libtriton.pyd python/triton/_C/libtriton.pyd
python/triton/_C/libtriton.so python/triton/_C/libtriton.so
.vscode
.vs

10
CMakeLists.txt
View File

@@ -15,8 +15,8 @@ if(NOT WIN32)
endif() endif()
# Options # Options
option(BUILD_TUTORIALS "Build C++ Triton tutorials" ON) option(TRITON_BUILD_TUTORIALS "Build C++ Triton tutorials" ON)
option(BUILD_PYTHON_MODULE "Build Python Triton bindings" OFF) option(TRITON_BUILD_PYTHON_MODULE "Build Python Triton bindings" OFF)
# Default build type # Default build type
if(NOT CMAKE_BUILD_TYPE) if(NOT CMAKE_BUILD_TYPE)
@@ -128,7 +128,7 @@ endif()
include_directories(${LLVM_INCLUDE_DIRS}) include_directories(${LLVM_INCLUDE_DIRS})
# Python module # Python module
if(BUILD_PYTHON_MODULE) if(TRITON_BUILD_PYTHON_MODULE)
message(STATUS "Adding Python module") message(STATUS "Adding Python module")
set(PYTHON_SRC_PATH ${CMAKE_CURRENT_SOURCE_DIR}/python/src) set(PYTHON_SRC_PATH ${CMAKE_CURRENT_SOURCE_DIR}/python/src)
include_directories("." ${PYTHON_SRC_PATH} ${PYTHON_INCLUDE_DIRS}) include_directories("." ${PYTHON_SRC_PATH} ${PYTHON_INCLUDE_DIRS})
@@ -139,7 +139,7 @@ endif()
# # Triton # # Triton
# file(GLOB_RECURSE LIBTRITON_SRC lib/*.cc) # file(GLOB_RECURSE LIBTRITON_SRC lib/*.cc)
# if (WIN32 AND BUILD_PYTHON_MODULE) # if (WIN32 AND TRITON_BUILD_PYTHON_MODULE)
# find_package(Python3 REQUIRED COMPONENTS Development) # find_package(Python3 REQUIRED COMPONENTS Development)
# Python3_add_library(triton SHARED ${LIBTRITON_SRC} ${PYTHON_SRC}) # Python3_add_library(triton SHARED ${LIBTRITON_SRC} ${PYTHON_SRC})
# set_target_properties(triton PROPERTIES SUFFIX ".pyd") # set_target_properties(triton PROPERTIES SUFFIX ".pyd")
@@ -209,7 +209,7 @@ else()
endif() endif()
if(BUILD_PYTHON_MODULE AND NOT WIN32) if(TRITON_BUILD_PYTHON_MODULE AND NOT WIN32)
set(CMAKE_SHARED_LIBRARY_SUFFIX ".so") set(CMAKE_SHARED_LIBRARY_SUFFIX ".so")
# Check if the platform is MacOS # Check if the platform is MacOS
if(APPLE) if(APPLE)

4
python/setup.py
View File

@@ -97,8 +97,8 @@ class CMakeBuild(build_ext):
python_include_dirs = [distutils.sysconfig.get_python_inc()] + ['/usr/local/cuda/include'] python_include_dirs = [distutils.sysconfig.get_python_inc()] + ['/usr/local/cuda/include']
cmake_args = [ cmake_args = [
"-DCMAKE_LIBRARY_OUTPUT_DIRECTORY=" + extdir, "-DCMAKE_LIBRARY_OUTPUT_DIRECTORY=" + extdir,
"-DBUILD_TUTORIALS=OFF", "-DTRITON_BUILD_TUTORIALS=OFF",
"-DBUILD_PYTHON_MODULE=ON", "-DTRITON_BUILD_PYTHON_MODULE=ON",
"-DLLVM_INCLUDE_DIRS=" + llvm_include_dir, "-DLLVM_INCLUDE_DIRS=" + llvm_include_dir,
"-DLLVM_LIBRARY_DIR=" + llvm_library_dir, "-DLLVM_LIBRARY_DIR=" + llvm_library_dir,
# '-DPYTHON_EXECUTABLE=' + sys.executable, # '-DPYTHON_EXECUTABLE=' + sys.executable,

2
test/CMakeLists.txt
View File

@@ -18,7 +18,7 @@ set(TRITON_TEST_DEPENDS
add_lit_testsuite(check-triton "Running the triton regression tests" add_lit_testsuite(check-triton "Running the triton regression tests"
${CMAKE_CURRENT_BINARY_DIR} ${CMAKE_CURRENT_BINARY_DIR}
DEPENDS ${STANDALONE_TEST_DEPENDS} DEPENDS ${TRITON_TEST_DEPENDS}
) )
set_target_properties(check-triton PROPERTIES FOLDER "Tests") set_target_properties(check-triton PROPERTIES FOLDER "Tests")