CMakeLists.txt 1.44 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
    eliminate_common_subexpression.cpp
8
    decompose.cpp
Paul's avatar
Paul committed
9
    propagate_constant.cpp
Paul's avatar
Paul committed
10
    dead_code_elimination.cpp
Paul's avatar
Paul committed
11
    eliminate_allocation.cpp
12
    eliminate_contiguous.cpp
13
    eliminate_concat.cpp
14
    eliminate_identity.cpp
15
    eliminate_pad.cpp
Paul's avatar
Paul committed
16
    rewrite_batchnorm.cpp
Shucai Xiao's avatar
Shucai Xiao committed
17
    rewrite_rnn.cpp
18
    rewrite_pooling.cpp
Paul's avatar
Paul committed
19
    env.cpp
Paul's avatar
Paul committed
20
    generate.cpp
Paul's avatar
Paul committed
21
    instruction.cpp
Paul's avatar
Paul committed
22
    program.cpp
Shucai Xiao's avatar
Shucai Xiao committed
23
    quantization.cpp
24
    remap.cpp
Paul's avatar
Paul committed
25
    shape.cpp
Paul's avatar
Paul committed
26
    schedule.cpp
27
    serialize.cpp
28
    pass_manager.cpp
Paul's avatar
Paul committed
29
    simplify_algebra.cpp
Paul's avatar
Paul committed
30
    simplify_reshapes.cpp
31
    value.cpp
Shucai Xiao's avatar
Shucai Xiao committed
32
    json.cpp
mei-ye's avatar
mei-ye committed
33
34
    opt/memory_coloring.cpp
    opt/memory_coloring_impl.cpp
Paul's avatar
Paul committed
35
)
Paul's avatar
Paul committed
36
rocm_set_soversion(migraphx ${MIGRAPHX_SO_VERSION})
Paul's avatar
Paul committed
37
rocm_clang_tidy_check(migraphx)
Paul's avatar
Paul committed
38
rocm_install_targets(
Paul's avatar
Paul committed
39
  TARGETS migraphx
Paul's avatar
Paul committed
40
41
42
  INCLUDE
    ${CMAKE_CURRENT_SOURCE_DIR}/include
)
Paul's avatar
Paul committed
43

Paul's avatar
Paul committed
44
find_path(HALF_INCLUDE_DIR half.hpp)
Paul's avatar
Paul committed
45
# TODO: Fix the incorrect path
Paul's avatar
Paul committed
46
target_include_directories(migraphx SYSTEM PUBLIC $<BUILD_INTERFACE:${HALF_INCLUDE_DIR}>)
Paul's avatar
Paul committed
47
48

set(PACKAGE_DEPENDS)
Paul's avatar
Paul committed
49

Paul Fultz II's avatar
Paul Fultz II committed
50
add_subdirectory(api)
Paul's avatar
Paul committed
51
add_subdirectory(driver)
Paul's avatar
Paul committed
52
add_subdirectory(onnx)
Khalique's avatar
Khalique committed
53
add_subdirectory(tf)
Khalique's avatar
Khalique committed
54

Paul's avatar
Paul committed
55
add_subdirectory(py)
Paul's avatar
Paul committed
56
add_subdirectory(targets/cpu)
Paul's avatar
Paul committed
57
if(MIGRAPHX_ENABLE_GPU)
Paul Fultz II's avatar
Paul Fultz II committed
58
list(APPEND PACKAGE_DEPENDS PACKAGE MIOpen PACKAGE rocblas)
Paul's avatar
Paul committed
59
add_subdirectory(targets/gpu)
Paul's avatar
Paul committed
60
endif()
mei-ye's avatar
mei-ye committed
61

Paul's avatar
Paul committed
62
rocm_export_targets(
Paul's avatar
Paul committed
63
64
  TARGETS migraphx::migraphx
  NAMESPACE migraphx::
Paul's avatar
Paul committed
65
66
67
  DEPENDS
    ${PACKAGE_DEPENDS}
)
mei-ye's avatar
mei-ye committed
68
69