CMakeLists.txt 919 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
    fwd_conv_batchnorm_rewrite.cpp
Paul's avatar
Paul committed
10
    env.cpp
Paul's avatar
Paul committed
11
    generate.cpp
Paul's avatar
Paul committed
12
    instruction.cpp
Paul's avatar
Paul committed
13
14
    program.cpp
    shape.cpp
Paul's avatar
Paul committed
15
    simplify_algebra.cpp
Paul's avatar
Paul committed
16
    simplify_reshapes.cpp
mei-ye's avatar
mei-ye committed
17
18
    opt/memory_coloring.cpp
    opt/memory_coloring_impl.cpp
Paul's avatar
Paul committed
19
)
Paul's avatar
Paul committed
20
21
rocm_clang_tidy_check(migraph)
target_include_directories(migraph PUBLIC $<BUILD_INTERFACE:${CMAKE_CURRENT_SOURCE_DIR}/include>)
Paul's avatar
Paul committed
22

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

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

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

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