CMakeLists.txt 1005 Bytes
Newer Older
Paul's avatar
Paul committed
1
2
3
4

list(APPEND CMAKE_PREFIX_PATH /opt/rocm /opt/rocm/hip /opt/rocm/hcc)
find_package(miopen)

5
6
7
8
# rocblas
find_package(rocblas REQUIRED PATHS /opt/rocm)
message(STATUS "Build with rocblas")

Paul's avatar
Paul committed
9
10
11
12
if(NOT TARGET MIOpen)
    message(SEND_ERROR "Cant find miopen")
endif()

13
add_library(migraph_device 
14
    device/contiguous.cpp
15
16
17
18
)
rocm_clang_tidy_check(migraph_device)
target_link_libraries(migraph_device migraph hip::device)
target_include_directories(migraph_device PUBLIC $<BUILD_INTERFACE:${CMAKE_CURRENT_SOURCE_DIR}/include>)
19
target_include_directories(migraph_device PRIVATE $<BUILD_INTERFACE:${CMAKE_CURRENT_SOURCE_DIR}/device>)
20

Paul's avatar
Paul committed
21
add_library(migraph_gpu
22
    eliminate_allocation.cpp
23
    eliminate_workspace.cpp
Paul's avatar
Paul committed
24
    hip.cpp
Paul's avatar
Paul committed
25
    target.cpp
Paul's avatar
Paul committed
26
    lowering.cpp
Paul's avatar
Paul committed
27
    write_literals.cpp
28
    rocblas.cpp
Paul's avatar
Paul committed
29
)
Paul's avatar
Paul committed
30
31
32
rocm_clang_tidy_check(migraph_gpu)
target_link_libraries(migraph_gpu migraph MIOpen migraph_device roc::rocblas)
target_include_directories(migraph_gpu PUBLIC $<BUILD_INTERFACE:${CMAKE_CURRENT_SOURCE_DIR}/include>)