CMakeLists.txt 868 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
    hip_contiguous.cpp
15
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
add_library(migraph_gpu
21
    eliminate_workspace.cpp
Paul's avatar
Paul committed
22
    hip.cpp
Paul's avatar
Paul committed
23
    target.cpp
Paul's avatar
Paul committed
24
    lowering.cpp
Paul's avatar
Paul committed
25
    write_literals.cpp
26
    rocblas.cpp
Paul's avatar
Paul committed
27
)
Paul's avatar
Paul committed
28
29
30
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>)