CMakeLists.txt 1.24 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
    common_subexpression_elimination.cpp
Paul's avatar
Paul committed
8
    constant_propagate.cpp
Paul's avatar
Paul committed
9
    dead_code_elimination.cpp
Paul's avatar
Paul committed
10
    eliminate_allocation.cpp
11
    eliminate_contiguous.cpp
12
    eliminate_concat.cpp
13
    eliminate_identity.cpp
14
    eliminate_pad.cpp
15
    fwd_conv_batchnorm_rewrite.cpp
Shucai Xiao's avatar
Shucai Xiao committed
16
    rewrite_rnn.cpp
Paul's avatar
Paul committed
17
    env.cpp
Paul's avatar
Paul committed
18
    generate.cpp
Paul's avatar
Paul committed
19
    instruction.cpp
Paul's avatar
Paul committed
20
    program.cpp
21
    quantize_ins.cpp
Paul's avatar
Paul committed
22
    shape.cpp
Paul's avatar
Paul committed
23
    schedule.cpp
24
    pass_manager.cpp
Paul's avatar
Paul committed
25
    simplify_algebra.cpp
Paul's avatar
Paul committed
26
    simplify_reshapes.cpp
mei-ye's avatar
mei-ye committed
27
28
    opt/memory_coloring.cpp
    opt/memory_coloring_impl.cpp
Paul's avatar
Paul committed
29
)
Paul's avatar
Paul committed
30
rocm_clang_tidy_check(migraphx)
Paul's avatar
Paul committed
31
rocm_install_targets(
Paul's avatar
Paul committed
32
  TARGETS migraphx
Paul's avatar
Paul committed
33
34
35
  INCLUDE
    ${CMAKE_CURRENT_SOURCE_DIR}/include
)
Paul's avatar
Paul committed
36

Paul's avatar
Paul committed
37
find_path(HALF_INCLUDE_DIR half.hpp)
Paul's avatar
Paul committed
38
# TODO: Fix the incorrect path
Paul's avatar
Paul committed
39
target_include_directories(migraphx SYSTEM PUBLIC $<BUILD_INTERFACE:${HALF_INCLUDE_DIR}>)
Paul's avatar
Paul committed
40
41

set(PACKAGE_DEPENDS)
Paul's avatar
Paul committed
42

Paul's avatar
Paul committed
43
add_subdirectory(onnx)
Khalique's avatar
Khalique committed
44
add_subdirectory(tf)
Khalique's avatar
Khalique committed
45

Paul's avatar
Paul committed
46
add_subdirectory(py)
Paul's avatar
Paul committed
47
add_subdirectory(targets/cpu)
Paul's avatar
Paul committed
48
if(MIGRAPHX_ENABLE_GPU)
Paul's avatar
Paul committed
49
list(APPEND PACKAGE_DEPENDS MIOpen rocblas)
Paul's avatar
Paul committed
50
add_subdirectory(targets/gpu)
Paul's avatar
Paul committed
51
endif()
mei-ye's avatar
mei-ye committed
52

Paul's avatar
Paul committed
53
rocm_export_targets(
Paul's avatar
Paul committed
54
55
  TARGETS migraphx::migraphx
  NAMESPACE migraphx::
Paul's avatar
Paul committed
56
57
58
  DEPENDS
    ${PACKAGE_DEPENDS}
)
mei-ye's avatar
mei-ye committed
59
60