CMakeLists.txt 1.22 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
    propagate_constant.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
21
    program.cpp
    shape.cpp
Paul's avatar
Paul committed
22
    schedule.cpp
23
    pass_manager.cpp
Paul's avatar
Paul committed
24
    simplify_algebra.cpp
Paul's avatar
Paul committed
25
    simplify_reshapes.cpp
mei-ye's avatar
mei-ye committed
26
27
    opt/memory_coloring.cpp
    opt/memory_coloring_impl.cpp
Paul's avatar
Paul committed
28
)
Paul's avatar
Paul committed
29
rocm_clang_tidy_check(migraphx)
Paul's avatar
Paul committed
30
rocm_install_targets(
Paul's avatar
Paul committed
31
  TARGETS migraphx
Paul's avatar
Paul committed
32
33
34
  INCLUDE
    ${CMAKE_CURRENT_SOURCE_DIR}/include
)
Paul's avatar
Paul committed
35

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

set(PACKAGE_DEPENDS)
Paul's avatar
Paul committed
41

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

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

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