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

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

set(PACKAGE_DEPENDS)
Paul's avatar
Paul committed
48

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

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

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