CMakeLists.txt 1.01 KB
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 
Paul's avatar
Paul committed
14
    device/add_relu.cpp
15
    device/contiguous.cpp
16
17
18
19
)
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>)
Paul's avatar
Paul committed
20
target_include_directories(migraph_device PRIVATE $<BUILD_INTERFACE:${CMAKE_CURRENT_SOURCE_DIR}/device/include>)
21

Paul's avatar
Paul committed
22
add_library(migraph_gpu
23
    eliminate_allocation.cpp
24
    eliminate_workspace.cpp
Paul's avatar
Paul committed
25
    hip.cpp
Paul's avatar
Paul committed
26
    target.cpp
Paul's avatar
Paul committed
27
    lowering.cpp
Paul's avatar
Paul committed
28
    write_literals.cpp
29
    rocblas.cpp
Paul's avatar
Paul committed
30
)
Paul's avatar
Paul committed
31
32
33
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>)