|
| 1 | +include(FetchContent) |
| 2 | + |
| 3 | +set(LLVM_DOWNLOAD_URL https://paddle-inference-dist.bj.bcebos.com/CINN/llvm11.tar.gz) |
| 4 | +set(LLVM_MD5 39d32b6be466781dddf5869318dcba53) |
| 5 | + |
| 6 | +set(FETCHCONTENT_BASE_DIR ${THIRD_PARTY_PATH}/llvm) |
| 7 | +set(FETCHCONTENT_QUIET OFF) |
| 8 | +FetchContent_Declare(external_llvm |
| 9 | + URL ${LLVM_DOWNLOAD_URL} |
| 10 | + URL_MD5 ${LLVM_MD5} |
| 11 | + PREFIX ${THIRD_PARTY_PATH}/llvm |
| 12 | + SOURCE_DIR ${THIRD_PARTY_PATH}/install/llvm |
| 13 | +) |
| 14 | +if (NOT LLVM_PATH) |
| 15 | + FetchContent_GetProperties(external_llvm) |
| 16 | + if (NOT external_llvm_POPULATED) |
| 17 | + FetchContent_Populate(external_llvm) |
| 18 | + endif() |
| 19 | + set(LLVM_PATH ${THIRD_PARTY_PATH}/install/llvm) |
| 20 | + set(LLVM_DIR ${THIRD_PARTY_PATH}/install/llvm/lib/cmake/llvm) |
| 21 | + set(MLIR_DIR ${THIRD_PARTY_PATH}/install/llvm/lib/cmake/mlir) |
| 22 | +else () |
| 23 | + set(LLVM_DIR ${LLVM_PATH}/lib/cmake/llvm) |
| 24 | + set(MLIR_DIR ${LLVM_PATH}/lib/cmake/mlir) |
| 25 | +endif() |
| 26 | + |
| 27 | +if (${CMAKE_CXX_COMPILER} STREQUAL "clang++") |
| 28 | + set(CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} -stdlib=libc++ -lc++abi") |
| 29 | +endif() |
| 30 | + |
| 31 | +message(STATUS "set LLVM_DIR: ${LLVM_DIR}") |
| 32 | +message(STATUS "set MLIR_DIR: ${MLIR_DIR}") |
| 33 | +find_package(LLVM REQUIRED CONFIG HINTS ${LLVM_DIR}) |
| 34 | +find_package(MLIR REQUIRED CONFIG HINTS ${MLIR_DIR}) |
| 35 | +find_package(ZLIB REQUIRED) |
| 36 | + |
| 37 | +list(APPEND CMAKE_MODULE_PATH "${LLVM_CMAKE_DIR}") |
| 38 | +include(AddLLVM) |
| 39 | + |
| 40 | +include_directories(${LLVM_INCLUDE_DIRS}) |
| 41 | +list(APPEND CMAKE_MODULE_PATH "${LLVM_CMAKE_DIR}") |
| 42 | +list(APPEND CMAKE_MODULE_PATH "${MLIR_CMAKE_DIR}") |
| 43 | +include(AddLLVM) |
| 44 | +include(TableGen) |
| 45 | +include(AddMLIR) |
| 46 | + |
| 47 | +message(STATUS "Found MLIR: ${MLIR_DIR}") |
| 48 | +message(STATUS "Found LLVM ${LLVM_PACKAGE_VERSION}") |
| 49 | +message(STATUS "Using LLVMConfig.cmake in: ${LLVM_DIR}") |
| 50 | + |
| 51 | +# To build with MLIR, the LLVM is build from source code using the following flags: |
| 52 | + |
| 53 | +#[==[ |
| 54 | +cmake -G Ninja ../llvm \ |
| 55 | + -DLLVM_ENABLE_PROJECTS="mlir;clang" \ |
| 56 | + -DLLVM_BUILD_EXAMPLES=OFF \ |
| 57 | + -DLLVM_TARGETS_TO_BUILD="X86" \ |
| 58 | + -DCMAKE_BUILD_TYPE=Release \ |
| 59 | + -DLLVM_ENABLE_ASSERTIONS=ON \ |
| 60 | + -DLLVM_ENABLE_ZLIB=OFF \ |
| 61 | + -DLLVM_ENABLE_RTTI=ON \ |
| 62 | +#]==] |
| 63 | +# The matched llvm-project version is f9dc2b7079350d0fed3bb3775f496b90483c9e42 (currently a temporary commit) |
| 64 | + |
| 65 | +add_definitions(${LLVM_DEFINITIONS}) |
| 66 | + |
| 67 | +llvm_map_components_to_libnames(llvm_libs Support Core irreader |
| 68 | + X86 executionengine orcjit mcjit all codegen) |
| 69 | + |
| 70 | +message(STATUS "LLVM libs: ${llvm_libs}") |
| 71 | + |
| 72 | +get_property(mlir_libs GLOBAL PROPERTY MLIR_ALL_LIBS) |
| 73 | +message(STATUS "MLIR libs: ${mlir_libs}") |
| 74 | +add_definitions(${LLVM_DEFINITIONS}) |
| 75 | + |
| 76 | + |
| 77 | +# The minimum needed libraries for MLIR IR parse and transform. |
| 78 | +set(MLIR_IR_LIBS MLIRAnalysis MLIRStandardOps MLIRPass MLIRParser MLIRDialect MLIRIR MLIROptLib) |
| 79 | + |
| 80 | + |
| 81 | +# tb_base is the name of a xxx.td file (without the .td suffix) |
| 82 | +function(mlir_tablegen_on td_base) |
| 83 | + set(options) |
| 84 | + set(oneValueArgs DIALECT) |
| 85 | + cmake_parse_arguments(mlir_tablegen_on "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN}) |
| 86 | + |
| 87 | + set(LLVM_TARGET_DEFINITIONS ${td_base}.td) |
| 88 | + mlir_tablegen(${td_base}.hpp.inc -gen-op-decls) |
| 89 | + mlir_tablegen(${td_base}.cpp.inc -gen-op-defs) |
| 90 | + if (mlir_tablegen_on_DIALECT) |
| 91 | + mlir_tablegen(${td_base}_dialect.hpp.inc --gen-dialect-decls -dialect=${mlir_tablegen_on_DIALECT}) |
| 92 | + endif() |
| 93 | + add_public_tablegen_target(${td_base}_IncGen) |
| 94 | + add_custom_target(${td_base}_inc DEPENDS ${td_base}_IncGen) |
| 95 | +endfunction() |
| 96 | + |
| 97 | +function(mlir_add_rewriter td_base) |
| 98 | + set(LLVM_TARGET_DEFINITIONS ${td_base}.td) |
| 99 | + mlir_tablegen(${td_base}.hpp.inc -gen-rewriters "-I${CMAKE_SOURCE_DIR}/infrt/dialect/pass") |
| 100 | + add_public_tablegen_target(${td_base}_IncGen) |
| 101 | + add_custom_target(${td_base}_inc DEPENDS ${td_base}_IncGen) |
| 102 | +endfunction() |
| 103 | + |
| 104 | +# Execute the mlir script with infrt-exec program. |
| 105 | +# @name: name of the test |
| 106 | +# @script: path to the mlir script file |
| 107 | +function (infrt_exec_check name script) |
| 108 | + add_test(NAME ${name} |
| 109 | + COMMAND sh -c "${CMAKE_BINARY_DIR}/paddle/infrt/host_context/infrt-exec -i ${CMAKE_CURRENT_SOURCE_DIR}/${script}| ${LLVM_PATH}/bin/FileCheck ${CMAKE_CURRENT_SOURCE_DIR}/${script}") |
| 110 | +endfunction() |
0 commit comments