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

# rocblas
find_package(rocblas REQUIRED PATHS /opt/rocm)
message(STATUS "Build with rocblas")

if(NOT TARGET MIOpen)
    message(SEND_ERROR "Cant find miopen")
endif()

add_library(migraph_device
    device/add.cpp
    device/add_relu.cpp
    device/contiguous.cpp
    device/mul.cpp
    device/concat.cpp
)
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>)
target_include_directories(migraph_device PRIVATE $<BUILD_INTERFACE:${CMAKE_CURRENT_SOURCE_DIR}/device/include>)

add_library(migraph_gpu
    eliminate_workspace.cpp
    fuse_ops.cpp
    hip.cpp
    target.cpp
    lowering.cpp
    gemm.cpp
    pooling.cpp
    convolution.cpp
    softmax.cpp
    contiguous.cpp
    concat.cpp
    relu.cpp
    leaky_relu.cpp
    add.cpp
    mul.cpp
    batchnorm.cpp
    write_literals.cpp
    rocblas.cpp
)
rocm_clang_tidy_check(migraph_gpu)
target_link_libraries(migraph_gpu PUBLIC migraph MIOpen roc::rocblas)
target_link_libraries(migraph_gpu PRIVATE migraph_device)

rocm_install_targets(
  TARGETS migraph_gpu migraph_device
  INCLUDE
    ${CMAKE_CURRENT_SOURCE_DIR}/include
)