CMakeLists.txt 1.14 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 
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_allocation.cpp
22
    eliminate_workspace.cpp
Paul's avatar
Paul committed
23
    hip.cpp
Paul's avatar
Paul committed
24
    target.cpp
Paul's avatar
Paul committed
25
    lowering.cpp
Paul's avatar
Paul committed
26
    write_literals.cpp
27
    rocblas.cpp
Paul's avatar
Paul committed
28
)
Paul's avatar
Paul committed
29
30
31
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>)
mei-ye's avatar
staging  
mei-ye committed
32
33
34
35
36
37
38
39

install (TARGETS migraph_gpu
  LIBRARY DESTINATION /opt/rocm/lib)
install (DIRECTORY include/migraph DESTINATION /opt/rocm/include)

install (TARGETS migraph_device
  LIBRARY DESTINATION /opt/rocm/lib)
install (DIRECTORY include/migraph DESTINATION /opt/rocm/include)