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
    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
    program.cpp
Shucai Xiao's avatar
Shucai Xiao committed
21
    quantization.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