2019-03-30 06:10:12 +01:00
|
|
|
# MLIR project.
|
2019-07-24 05:09:11 +02:00
|
|
|
set(MLIR_MAIN_SRC_DIR ${CMAKE_CURRENT_SOURCE_DIR}/include ) # --src-root
|
2019-03-30 06:10:12 +01:00
|
|
|
set(MLIR_INCLUDE_DIR ${CMAKE_CURRENT_BINARY_DIR}/include ) # --includedir
|
|
|
|
set(MLIR_TABLEGEN_EXE mlir-tblgen)
|
|
|
|
|
|
|
|
set(MLIR_SOURCE_DIR ${CMAKE_CURRENT_SOURCE_DIR})
|
|
|
|
set(MLIR_BINARY_DIR ${CMAKE_CURRENT_BINARY_DIR})
|
|
|
|
|
|
|
|
function(mlir_tablegen ofn)
|
|
|
|
tablegen(MLIR ${ARGV} "-I${MLIR_MAIN_SRC_DIR}" "-I${MLIR_INCLUDE_DIR}")
|
|
|
|
set(TABLEGEN_OUTPUT ${TABLEGEN_OUTPUT} ${CMAKE_CURRENT_BINARY_DIR}/${ofn}
|
|
|
|
PARENT_SCOPE)
|
|
|
|
endfunction()
|
|
|
|
|
2019-12-19 22:16:59 +01:00
|
|
|
function(add_mlir_dialect dialect dialect_doc_filename)
|
2019-12-02 18:17:51 +01:00
|
|
|
set(LLVM_TARGET_DEFINITIONS ${dialect}.td)
|
|
|
|
mlir_tablegen(${dialect}.h.inc -gen-op-decls)
|
|
|
|
mlir_tablegen(${dialect}.cpp.inc -gen-op-defs)
|
|
|
|
add_public_tablegen_target(MLIR${dialect}IncGen)
|
|
|
|
|
|
|
|
# Generate Dialect Documentation
|
2019-12-19 22:16:59 +01:00
|
|
|
set(LLVM_TARGET_DEFINITIONS ${dialect_doc_filename}.td)
|
|
|
|
tablegen(MLIR ${dialect_doc_filename}.md -gen-op-doc "-I${MLIR_MAIN_SRC_DIR}" "-I${MLIR_INCLUDE_DIR}")
|
|
|
|
set(GEN_DOC_FILE ${MLIR_BINARY_DIR}/docs/Dialects/${dialect_doc_filename}.md)
|
2019-12-02 18:17:51 +01:00
|
|
|
add_custom_command(
|
|
|
|
OUTPUT ${GEN_DOC_FILE}
|
|
|
|
COMMAND ${CMAKE_COMMAND} -E copy
|
2019-12-19 22:16:59 +01:00
|
|
|
${CMAKE_CURRENT_BINARY_DIR}/${dialect_doc_filename}.md
|
2019-12-02 18:17:51 +01:00
|
|
|
${GEN_DOC_FILE}
|
2019-12-19 22:16:59 +01:00
|
|
|
DEPENDS ${CMAKE_CURRENT_BINARY_DIR}/${dialect_doc_filename}.md)
|
|
|
|
add_custom_target(${dialect_doc_filename}DocGen DEPENDS ${GEN_DOC_FILE})
|
|
|
|
add_dependencies(mlir-doc ${dialect_doc_filename}DocGen)
|
2019-12-02 18:17:51 +01:00
|
|
|
endfunction()
|
|
|
|
|
|
|
|
add_custom_target(mlir-doc)
|
|
|
|
|
2019-03-30 06:10:12 +01:00
|
|
|
# TODO: This is to handle the current static registration, but should be
|
|
|
|
# factored out a bit.
|
|
|
|
function(whole_archive_link target)
|
2019-04-01 19:17:45 +02:00
|
|
|
if("${CMAKE_SYSTEM_NAME}" STREQUAL "Darwin")
|
2019-04-24 01:39:15 +02:00
|
|
|
set(link_flags "-L${CMAKE_BINARY_DIR}/lib ")
|
2019-04-01 19:17:45 +02:00
|
|
|
FOREACH(LIB ${ARGN})
|
2019-04-24 01:39:15 +02:00
|
|
|
string(CONCAT link_flags ${link_flags} "-Wl,-force_load ${CMAKE_BINARY_DIR}/lib/lib${LIB}.a ")
|
2019-04-01 19:17:45 +02:00
|
|
|
ENDFOREACH(LIB)
|
2019-05-29 22:47:24 +02:00
|
|
|
elseif(MSVC)
|
|
|
|
FOREACH(LIB ${ARGN})
|
2019-12-31 20:04:29 +01:00
|
|
|
string(CONCAT link_flags ${link_flags} "/WHOLEARCHIVE:lib/${LIB}.lib ")
|
2019-05-29 22:47:24 +02:00
|
|
|
ENDFOREACH(LIB)
|
2019-04-01 19:17:45 +02:00
|
|
|
else()
|
2019-04-20 19:45:59 +02:00
|
|
|
set(link_flags "-L${CMAKE_BINARY_DIR}/lib -Wl,--whole-archive,")
|
2019-04-01 19:17:45 +02:00
|
|
|
FOREACH(LIB ${ARGN})
|
|
|
|
string(CONCAT link_flags ${link_flags} "-l${LIB},")
|
|
|
|
ENDFOREACH(LIB)
|
|
|
|
string(CONCAT link_flags ${link_flags} "--no-whole-archive")
|
|
|
|
endif()
|
2019-03-30 06:10:12 +01:00
|
|
|
set_target_properties(${target} PROPERTIES LINK_FLAGS ${link_flags})
|
|
|
|
endfunction(whole_archive_link)
|
|
|
|
|
2019-06-26 14:16:11 +02:00
|
|
|
# Build the CUDA conversions and run according tests if the NVPTX backend
|
|
|
|
# is available
|
|
|
|
if ("NVPTX" IN_LIST LLVM_TARGETS_TO_BUILD)
|
|
|
|
set(MLIR_CUDA_CONVERSIONS_ENABLED 1)
|
|
|
|
else()
|
|
|
|
set(MLIR_CUDA_CONVERSIONS_ENABLED 0)
|
|
|
|
endif()
|
|
|
|
|
2019-07-04 16:49:52 +02:00
|
|
|
set(MLIR_CUDA_RUNNER_ENABLED 0 CACHE BOOL "Enable building the mlir CUDA runner")
|
|
|
|
|
2019-03-30 06:10:12 +01:00
|
|
|
include_directories( "include")
|
|
|
|
include_directories( ${MLIR_INCLUDE_DIR})
|
|
|
|
|
|
|
|
add_subdirectory(include/mlir)
|
|
|
|
add_subdirectory(lib)
|
|
|
|
add_subdirectory(tools)
|
|
|
|
add_subdirectory(unittests)
|
|
|
|
add_subdirectory(test)
|
2019-04-02 19:02:07 +02:00
|
|
|
|
|
|
|
if( LLVM_INCLUDE_EXAMPLES )
|
|
|
|
add_subdirectory(examples)
|
|
|
|
endif()
|
2019-11-20 06:04:45 +01:00
|
|
|
|
|
|
|
if (NOT LLVM_INSTALL_TOOLCHAIN_ONLY)
|
|
|
|
install(DIRECTORY include/mlir include/mlir-c
|
|
|
|
DESTINATION include
|
|
|
|
COMPONENT mlir-headers
|
|
|
|
FILES_MATCHING
|
|
|
|
PATTERN "*.h"
|
|
|
|
PATTERN "*.inc"
|
2019-12-29 18:04:09 +01:00
|
|
|
PATTERN "*.td"
|
2019-11-20 06:04:45 +01:00
|
|
|
PATTERN "LICENSE.TXT"
|
|
|
|
)
|
|
|
|
|
|
|
|
install(DIRECTORY ${MLIR_INCLUDE_DIR}/mlir ${MLIR_INCLUDE_DIR}/mlir-c
|
|
|
|
DESTINATION include
|
|
|
|
COMPONENT mlir-headers
|
|
|
|
FILES_MATCHING
|
|
|
|
PATTERN "*.h"
|
|
|
|
PATTERN "*.gen"
|
|
|
|
PATTERN "*.inc"
|
|
|
|
PATTERN "CMakeFiles" EXCLUDE
|
|
|
|
PATTERN "config.h" EXCLUDE
|
|
|
|
)
|
|
|
|
|
|
|
|
if (NOT LLVM_ENABLE_IDE)
|
|
|
|
add_llvm_install_targets(install-mlir-headers
|
|
|
|
DEPENDS mlir-headers
|
|
|
|
COMPONENT mlir-headers)
|
|
|
|
endif()
|
|
|
|
endif()
|