CMakeLists.txt 1.14 KB
Newer Older
Paul's avatar
Paul committed
1

Paul's avatar
Paul committed
2
add_library(migraphx_cpu
3
4
5
6
7
8
9
10
11
12
13
    allocation_model.cpp
    allocate.cpp
    add.cpp
    contiguous.cpp
    concat.cpp
    convolution.cpp
    copy.cpp
    mul.cpp
    pooling.cpp
    relu.cpp
    gemm.cpp
Shucai Xiao's avatar
Shucai Xiao committed
14
15
    target.cpp
    lowering.cpp
Paul's avatar
Paul committed
16
)
Paul's avatar
Paul committed
17
set_target_properties(migraphx_cpu PROPERTIES EXPORT_NAME cpu)
Paul's avatar
Paul committed
18
rocm_set_soversion(migraphx_cpu ${MIGRAPHX_SO_VERSION})
Paul's avatar
Paul committed
19

Paul's avatar
Paul committed
20
find_package(Threads)
21
find_package(dnnl REQUIRED)
Paul's avatar
Paul committed
22

Paul's avatar
Paul committed
23
rocm_clang_tidy_check(migraphx_cpu)
24
25
26
27
28
29
30
31
32
33
34
target_link_libraries(migraphx_cpu PRIVATE migraphx Threads::Threads)
target_link_libraries(migraphx_cpu PRIVATE DNNL::dnnl)
find_package(OpenMP)
target_link_libraries(migraphx_cpu PUBLIC OpenMP::OpenMP_CXX)
# Add library path to rpath to workaround issues with our broken packages
foreach(LIBRARY ${OpenMP_CXX_LIBRARIES})
    if(LIBRARY MATCHES "libomp")
        get_filename_component(LIBRARY_PATH "${LIBRARY}" PATH)
        target_link_libraries(migraphx_cpu PUBLIC -Wl,-rpath=${LIBRARY_PATH} -Wl,-rpath-link=${LIBRARY_PATH})
    endif()
endforeach()
mei-ye's avatar
mei-ye committed
35

36
37
target_link_libraries(migraphx_all_targets INTERFACE migraphx_cpu)

Paul's avatar
Paul committed
38
rocm_install_targets(
Paul's avatar
Paul committed
39
  TARGETS migraphx_cpu
Paul's avatar
Paul committed
40
41
42
  INCLUDE
    ${CMAKE_CURRENT_SOURCE_DIR}/include
)
mei-ye's avatar
mei-ye committed
43