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

[BUILD] When set, use MLIR_DIR for finding both MLIR and LLVM (#755)

Browse Source
This commit is contained in:
Ian Bearman
2022-10-09 13:11:20 -07:00
committed by GitHub
parent 89f6e1db5e
commit ccc5ab6ac9
1 changed files with 64 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
CMakeLists.txt
View File

@@ -49,7 +49,8 @@ endif()
########## ##########
# LLVM # LLVM
########## ##########
if("${LLVM_LIBRARY_DIR}" STREQUAL "") if (NOT MLIR_DIR)
if(NOT LLVM_LIBRARY_DIR)
if(WIN32) if(WIN32)
find_package(LLVM 13 REQUIRED COMPONENTS nvptx amdgpu) find_package(LLVM 13 REQUIRED COMPONENTS nvptx amdgpu)
@@ -127,7 +128,8 @@ libLLVMCoroutines.a
libLLVMAnalysis.a libLLVMAnalysis.a
) )
endif() endif()
include_directories(${LLVM_INCLUDE_DIRS}) set (MLIR_DIR ${LLVM_LIBRARY_DIR}/cmake/mlir)
endif()
# Python module # Python module
if(TRITON_BUILD_PYTHON_MODULE) if(TRITON_BUILD_PYTHON_MODULE)
@@ -161,10 +163,10 @@ endif()
# MLIR # MLIR
find_package(MLIR REQUIRED CONFIG PATHS ${LLVM_LIBRARY_DIR}/cmake/mlir) find_package(MLIR REQUIRED CONFIG PATHS ${MLIR_DIR})
list(APPEND CMAKE_MODULE_PATH ${LLVM_LIBRARY_DIR}/cmake/llvm) list(APPEND CMAKE_MODULE_PATH "${MLIR_CMAKE_DIR}")
list(APPEND CMAKE_MODULE_PATH ${LLVM_LIBRARY_DIR}/cmake/mlir) list(APPEND CMAKE_MODULE_PATH "${LLVM_CMAKE_DIR}")
include(TableGen) # required by AddMLIR include(TableGen) # required by AddMLIR
include(AddLLVM) include(AddLLVM)
@@ -172,6 +174,7 @@ include(AddMLIR)
# include(HandleLLVMOptions) # human-friendly error message # include(HandleLLVMOptions) # human-friendly error message
include_directories(${MLIR_INCLUDE_DIRS}) include_directories(${MLIR_INCLUDE_DIRS})
include_directories(${LLVM_INCLUDE_DIRS})
include_directories(${PROJECT_SOURCE_DIR}/include) include_directories(${PROJECT_SOURCE_DIR}/include)
include_directories(${PROJECT_BINARY_DIR}/include) # Tablegen'd files include_directories(${PROJECT_BINARY_DIR}/include) # Tablegen'd files
# link_directories(${LLVM_LIBRARY_DIR}) # link_directories(${LLVM_LIBRARY_DIR})