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
    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
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