CMakeLists.txt 1.39 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
    pass_manager.cpp
Paul's avatar
Paul committed
28
    simplify_algebra.cpp
Paul's avatar
Paul committed
29
    simplify_reshapes.cpp
mei-ye's avatar
mei-ye committed
30
31
    opt/memory_coloring.cpp
    opt/memory_coloring_impl.cpp
Paul's avatar
Paul committed
32
)
Paul's avatar
Paul committed
33
rocm_set_soversion(migraphx ${MIGRAPHX_SO_VERSION})
Paul's avatar
Paul committed
34
rocm_clang_tidy_check(migraphx)
Paul's avatar
Paul committed
35
rocm_install_targets(
Paul's avatar
Paul committed
36
  TARGETS migraphx
Paul's avatar
Paul committed
37
38
39
  INCLUDE
    ${CMAKE_CURRENT_SOURCE_DIR}/include
)
Paul's avatar
Paul committed
40

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

set(PACKAGE_DEPENDS)
Paul's avatar
Paul committed
46

Paul Fultz II's avatar
Paul Fultz II committed
47
add_subdirectory(api)
Paul's avatar
Paul committed
48
add_subdirectory(driver)
Paul's avatar
Paul committed
49
add_subdirectory(onnx)
Khalique's avatar
Khalique committed
50
add_subdirectory(tf)
Khalique's avatar
Khalique committed
51

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

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