FluidML/CMakeLists.txt at master · sgjzfzzf/FluidML · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
cmake_minimum_required(VERSION 3.17)

project(CpuTransformers)
set(PYTHON_PROJECT_NAME "fluidml")

set(CMAKE_CXX_STANDARD_REQUIRED 17)
set(CMAKE_EXPORT_COMPILE_COMMANDS ON)

include(ExternalProject)

option(CMAKE_BUILD_TYPE "Build type" Debug)
option(BUILD_TESTS "Build tests" OFF)
option(BUILD_PYTHON "Build python bindings" OFF)
option(DP_DEBUG "Skip the Dynamic Programming optimization for debug" OFF)
option(USE_LOGS "Use logs" OFF)

ExternalProject_Add(
    libbenchmark
    GIT_REPOSITORY https://github.com/google/benchmark.git
    GIT_TAG v1.9.0
    CMAKE_ARGS -DBENCHMARK_DOWNLOAD_DEPENDENCIES=ON -DCMAKE_BUILD_TYPE=Release -DCMAKE_C_COMPILER=${CMAKE_C_COMPILER} -DCMAKE_CXX_COMPILER=${CMAKE_CXX_COMPILER} -DCMAKE_INSTALL_PREFIX=${CMAKE_BINARY_DIR}/external
    BUILD_COMMAND ${CMAKE_COMMAND} --build . --target all
    INSTALL_COMMAND ${CMAKE_COMMAND} --build . --target install
)

ExternalProject_Add(
    libfmt
    GIT_REPOSITORY https://github.com/fmtlib/fmt.git
    GIT_TAG 10.2.1
    CMAKE_ARGS -DCMAKE_BUILD_TYPE=Release -DCMAKE_INSTALL_PREFIX=${CMAKE_BINARY_DIR}/external -DCMAKE_C_COMPILER=${CMAKE_C_COMPILER} -DCMAKE_CXX_COMPILER=${CMAKE_CXX_COMPILER} -DCMAKE_BUILD_TYPE=Release
    BUILD_COMMAND ${CMAKE_COMMAND} --build . --target all
    INSTALL_COMMAND ${CMAKE_COMMAND} --build . --target install
)

ExternalProject_Add(
    libglog
    GIT_REPOSITORY https://github.com/google/glog.git
    GIT_TAG v0.7.1
    CMAKE_ARGS -DCMAKE_BUILD_TYPE=Release -DCMAKE_INSTALL_PREFIX=${CMAKE_BINARY_DIR}/external -DCMAKE_C_COMPILER=${CMAKE_C_COMPILER} -DCMAKE_CXX_COMPILER=${CMAKE_CXX_COMPILER} -DCMAKE_BUILD_TYPE=Release
    BUILD_COMMAND ${CMAKE_COMMAND} --build . --target all
    INSTALL_COMMAND ${CMAKE_COMMAND} --build . --target install
)

ExternalProject_Add(
    libjson
    GIT_REPOSITORY https://github.com/nlohmann/json.git
    GIT_TAG v3.11.3
    CMAKE_ARGS -DCMAKE_BUILD_TYPE=Release -DCMAKE_INSTALL_PREFIX=${CMAKE_BINARY_DIR}/external -DCMAKE_C_COMPILER=${CMAKE_C_COMPILER} -DCMAKE_CXX_COMPILER=${CMAKE_CXX_COMPILER} -DCMAKE_BUILD_TYPE=Release
    BUILD_COMMAND ${CMAKE_COMMAND} --build . --target all
    INSTALL_COMMAND ${CMAKE_COMMAND} --build . --target install
)

find_package(LLVM 18.1.6 REQUIRED CONFIG)

include_directories(include)
include_directories(${CMAKE_CURRENT_BINARY_DIR}/third_party/onnx)
include_directories(${CMAKE_CURRENT_SOURCE_DIR}/third_party/onnx)
include_directories(${LLVM_INCLUDE_DIRS})

link_directories(${CMAKE_CURRENT_BINARY_DIR}/third_party/onnx)
link_directories(${LLVM_LIBRARY_DIRS})

add_link_options(-Wl,-rpath,${LLVM_LIBRARY_DIRS})

add_subdirectory(bin)
add_subdirectory(third_party)
add_subdirectory(src)

if(${BUILD_TESTS})
    enable_testing()
    add_subdirectory(benchmark)
    add_subdirectory(test)
endif()

find_program(CLANG_FORMAT clang-format REQUIRED)
find_program(BLACK_FORMAT black REQUIRED)
add_custom_target(
    format
    COMMAND find ${CMAKE_SOURCE_DIR}/src ${CMAKE_SOURCE_DIR}/test ${CMAKE_SOURCE_DIR}/tools -name "\"*.py\"" -o -name "\"*.pyi\"" | xargs ${BLACK_FORMAT} ${CMAKE_SOURCE_DIR}/setup.py
    COMMAND find ${CMAKE_SOURCE_DIR}/bin ${CMAKE_SOURCE_DIR}/benchmark ${CMAKE_SOURCE_DIR}/include ${CMAKE_SOURCE_DIR}/src ${CMAKE_SOURCE_DIR}/test -name "\"*.cpp\"" -o -name "\"*.h\"" -o -name "\"*.hpp\"" | xargs ${CLANG_FORMAT} -i
)

install(
    DIRECTORY ${CMAKE_SOURCE_DIR}/include/
    DESTINATION include/fluidml
)