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

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

set(PACKAGE_DEPENDS)
Paul's avatar
Paul committed
37

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

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