CMakeLists.txt 944 Bytes
Newer Older
Paul's avatar
Paul committed
1

Paul's avatar
Paul committed
2
add_library(migraph 
Paul's avatar
Paul committed
3
    auto_contiguous.cpp
Paul's avatar
Paul committed
4
    common_subexpression_elimination.cpp
Paul's avatar
Paul committed
5
    constant_propagate.cpp
Paul's avatar
Paul committed
6
    dead_code_elimination.cpp
Paul's avatar
Paul committed
7
    eliminate_allocation.cpp
8
    eliminate_contiguous.cpp
9
    eliminate_concat.cpp
10
    fwd_conv_batchnorm_rewrite.cpp
Paul's avatar
Paul committed
11
    env.cpp
Paul's avatar
Paul committed
12
    generate.cpp
Paul's avatar
Paul committed
13
    instruction.cpp
Paul's avatar
Paul committed
14
15
    program.cpp
    shape.cpp
Paul's avatar
Paul committed
16
    simplify_algebra.cpp
Paul's avatar
Paul committed
17
    simplify_reshapes.cpp
mei-ye's avatar
mei-ye committed
18
19
    opt/memory_coloring.cpp
    opt/memory_coloring_impl.cpp
Paul's avatar
Paul committed
20
)
Paul's avatar
Paul committed
21
22
rocm_clang_tidy_check(migraph)
target_include_directories(migraph PUBLIC $<BUILD_INTERFACE:${CMAKE_CURRENT_SOURCE_DIR}/include>)
Paul's avatar
Paul committed
23

Paul's avatar
Paul committed
24
find_path(HALF_INCLUDE_DIR half.hpp)
Paul's avatar
Paul committed
25
target_include_directories(migraph SYSTEM PUBLIC ${HALF_INCLUDE_DIR})
Paul's avatar
Paul committed
26

Paul's avatar
Paul committed
27
add_subdirectory(onnx)
Paul's avatar
Paul committed
28
add_subdirectory(targets/cpu)
Paul's avatar
Paul committed
29
if(MIGRAPH_ENABLE_GPU)
Paul's avatar
Paul committed
30
add_subdirectory(targets/gpu)
Paul's avatar
Paul committed
31
endif()
mei-ye's avatar
mei-ye committed
32

mei-ye's avatar
mei-ye committed
33
34
#install (TARGETS migraph
#   LIBRARY DESTINATION /opt/rocm/lib)
mei-ye's avatar
mei-ye committed
35

mei-ye's avatar
mei-ye committed
36
#install (DIRECTORY include/migraph DESTINATION /opt/rocm/include)
mei-ye's avatar
mei-ye committed
37
38