CMakeLists.txt 1.66 KB
Newer Older
Paul's avatar
Paul committed
1

Paul's avatar
Paul committed
2
3
4
include(ROCMInstallTargets)
include(ROCMPackageConfigHelpers)

Paul's avatar
Paul committed
5
add_library(migraphx 
Paul's avatar
Paul committed
6
    auto_contiguous.cpp
Paul's avatar
Paul committed
7
    eliminate_common_subexpression.cpp
8
    decompose.cpp
Paul's avatar
Paul committed
9
    propagate_constant.cpp
Paul's avatar
Paul committed
10
    dead_code_elimination.cpp
Paul's avatar
Paul committed
11
    eliminate_allocation.cpp
12
    eliminate_contiguous.cpp
13
    eliminate_concat.cpp
14
    eliminate_identity.cpp
15
    eliminate_pad.cpp
Paul's avatar
Paul committed
16
    rewrite_batchnorm.cpp
Shucai Xiao's avatar
Shucai Xiao committed
17
    rewrite_rnn.cpp
18
    rewrite_pooling.cpp
Paul's avatar
Paul committed
19
    env.cpp
Paul's avatar
Paul committed
20
    generate.cpp
Paul's avatar
Paul committed
21
    instruction.cpp
22
    msgpack.cpp
Paul's avatar
Paul committed
23
    program.cpp
Shucai Xiao's avatar
Shucai Xiao committed
24
    quantization.cpp
25
    reduce_dims.cpp
26
    remap.cpp
Paul's avatar
Paul committed
27
    shape.cpp
Paul's avatar
Paul committed
28
    schedule.cpp
29
    serialize.cpp
30
    pass_manager.cpp
Paul's avatar
Paul committed
31
    simplify_algebra.cpp
Paul's avatar
Paul committed
32
    simplify_reshapes.cpp
33
    value.cpp
Shucai Xiao's avatar
Shucai Xiao committed
34
    json.cpp
mei-ye's avatar
mei-ye committed
35
36
    opt/memory_coloring.cpp
    opt/memory_coloring_impl.cpp
Paul's avatar
Paul committed
37
)
Paul's avatar
Paul committed
38
rocm_set_soversion(migraphx ${MIGRAPHX_SO_VERSION})
Paul's avatar
Paul committed
39
rocm_clang_tidy_check(migraphx)
Paul's avatar
Paul committed
40
rocm_install_targets(
Paul's avatar
Paul committed
41
  TARGETS migraphx
Paul's avatar
Paul committed
42
43
44
  INCLUDE
    ${CMAKE_CURRENT_SOURCE_DIR}/include
)
Paul's avatar
Paul committed
45

Paul's avatar
Paul committed
46
find_path(HALF_INCLUDE_DIR half.hpp)
Paul's avatar
Paul committed
47
# TODO: Fix the incorrect path
Paul's avatar
Paul committed
48
target_include_directories(migraphx SYSTEM PUBLIC $<BUILD_INTERFACE:${HALF_INCLUDE_DIR}>)
Paul's avatar
Paul committed
49

50
51
52
53
54
find_package(msgpack REQUIRED)
target_link_libraries(migraphx PRIVATE msgpackc-cxx)
# Make this available to the tests
target_link_libraries(migraphx INTERFACE $<BUILD_INTERFACE:msgpackc-cxx>)

Paul's avatar
Paul committed
55
set(PACKAGE_DEPENDS)
Paul's avatar
Paul committed
56

Paul Fultz II's avatar
Paul Fultz II committed
57
add_subdirectory(api)
Paul's avatar
Paul committed
58
add_subdirectory(driver)
Paul's avatar
Paul committed
59
add_subdirectory(onnx)
Khalique's avatar
Khalique committed
60
add_subdirectory(tf)
Khalique's avatar
Khalique committed
61

Paul's avatar
Paul committed
62
add_subdirectory(py)
Paul's avatar
Paul committed
63
add_subdirectory(targets/cpu)
Paul's avatar
Paul committed
64
if(MIGRAPHX_ENABLE_GPU)
Paul Fultz II's avatar
Paul Fultz II committed
65
list(APPEND PACKAGE_DEPENDS PACKAGE MIOpen PACKAGE rocblas)
Paul's avatar
Paul committed
66
add_subdirectory(targets/gpu)
Paul's avatar
Paul committed
67
endif()
mei-ye's avatar
mei-ye committed
68

Paul's avatar
Paul committed
69
rocm_export_targets(
Paul's avatar
Paul committed
70
71
  TARGETS migraphx::migraphx
  NAMESPACE migraphx::
Paul's avatar
Paul committed
72
73
74
  DEPENDS
    ${PACKAGE_DEPENDS}
)
mei-ye's avatar
mei-ye committed
75
76