CMakeLists.txt 1.46 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
    reduce_dims.cpp
25
    remap.cpp
Paul's avatar
Paul committed
26
    shape.cpp
Paul's avatar
Paul committed
27
    schedule.cpp
28
    serialize.cpp
29
    pass_manager.cpp
Paul's avatar
Paul committed
30
    simplify_algebra.cpp
Paul's avatar
Paul committed
31
    simplify_reshapes.cpp
32
    value.cpp
Shucai Xiao's avatar
Shucai Xiao committed
33
    json.cpp
mei-ye's avatar
mei-ye committed
34
35
    opt/memory_coloring.cpp
    opt/memory_coloring_impl.cpp
Paul's avatar
Paul committed
36
)
Paul's avatar
Paul committed
37
rocm_set_soversion(migraphx ${MIGRAPHX_SO_VERSION})
Paul's avatar
Paul committed
38
rocm_clang_tidy_check(migraphx)
Paul's avatar
Paul committed
39
rocm_install_targets(
Paul's avatar
Paul committed
40
  TARGETS migraphx
Paul's avatar
Paul committed
41
42
43
  INCLUDE
    ${CMAKE_CURRENT_SOURCE_DIR}/include
)
Paul's avatar
Paul committed
44

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

set(PACKAGE_DEPENDS)
Paul's avatar
Paul committed
50

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

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

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