CMakeLists.txt 1.09 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
    fwd_conv_batchnorm_rewrite.cpp
Paul's avatar
Paul committed
14
    env.cpp
Paul's avatar
Paul committed
15
    generate.cpp
Paul's avatar
Paul committed
16
    instruction.cpp
Paul's avatar
Paul committed
17
18
    program.cpp
    shape.cpp
Paul's avatar
Paul committed
19
    simplify_algebra.cpp
Paul's avatar
Paul committed
20
    simplify_reshapes.cpp
mei-ye's avatar
mei-ye committed
21
22
    opt/memory_coloring.cpp
    opt/memory_coloring_impl.cpp
Paul's avatar
Paul committed
23
)
Paul's avatar
Paul committed
24
rocm_clang_tidy_check(migraphx)
Paul's avatar
Paul committed
25
rocm_install_targets(
Paul's avatar
Paul committed
26
  TARGETS migraphx
Paul's avatar
Paul committed
27
28
29
  INCLUDE
    ${CMAKE_CURRENT_SOURCE_DIR}/include
)
Paul's avatar
Paul committed
30

Paul's avatar
Paul committed
31
find_path(HALF_INCLUDE_DIR half.hpp)
Paul's avatar
Paul committed
32
# TODO: Fix the incorrect path
Paul's avatar
Paul committed
33
target_include_directories(migraphx SYSTEM PUBLIC $<BUILD_INTERFACE:${HALF_INCLUDE_DIR}>)
Paul's avatar
Paul committed
34
35

set(PACKAGE_DEPENDS)
Paul's avatar
Paul committed
36

Paul's avatar
Paul committed
37
add_subdirectory(onnx)
Khalique's avatar
Khalique committed
38
add_subdirectory(tf)
Paul's avatar
Paul committed
39
add_subdirectory(targets/cpu)
Paul's avatar
Paul committed
40
if(MIGRAPHX_ENABLE_GPU)
Paul's avatar
Paul committed
41
list(APPEND PACKAGE_DEPENDS MIOpen rocblas)
Paul's avatar
Paul committed
42
add_subdirectory(targets/gpu)
Paul's avatar
Paul committed
43
endif()
mei-ye's avatar
mei-ye committed
44

Paul's avatar
Paul committed
45
rocm_export_targets(
Paul's avatar
Paul committed
46
47
  TARGETS migraphx::migraphx
  NAMESPACE migraphx::
Paul's avatar
Paul committed
48
49
50
  DEPENDS
    ${PACKAGE_DEPENDS}
)
mei-ye's avatar
mei-ye committed
51
52