cmake_minimum_required(VERSION 3.5)

set(CMAKE_INSTALL_PREFIX "/opt/rocm" CACHE PATH "")

project(miopentensile LANGUAGES CXX)

add_definitions(-std=c++14)

list(APPEND CMAKE_PREFIX_PATH ${CMAKE_INSTALL_PREFIX}/hcc ${CMAKE_INSTALL_PREFIX}/hip ${CMAKE_INSTALL_PREFIX}/llvm ${CMAKE_INSTALL_PREFIX} /opt/rocm/hcc /opt/rocm/hip /opt/rocm/llvm /opt/rocm)

execute_process (
    COMMAND bash -c "awk -F= '/^ID=/{print $2}' /etc/os-release |tr -d '\n' | tr -d '\"'"
    OUTPUT_VARIABLE outOS
)
if(outOS MATCHES "centos")
    if(EXISTS "/usr/lib64/llvm7.0")
        list(APPEND CMAKE_PREFIX_PATH /usr/lib64/llvm7.0)
    else()
        list(APPEND CMAKE_PREFIX_PATH /opt/rh/llvm-toolset-7/root/usr)
    endif()
endif()

find_package( ROCM CONFIG QUIET PATHS ${CMAKE_INSTALL_PREFIX} )
include(ROCMInstallTargets)
include(ROCMSetupVersion)

rocm_setup_version(VERSION 1.0)

set(CMAKE_LIBRARY_OUTPUT_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}/lib)
set(CMAKE_ARCHIVE_OUTPUT_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}/lib)
set(CMAKE_RUNTIME_OUTPUT_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}/bin)

set( AMDGPU_TARGETS "gfx803;gfx900;gfx906" CACHE STRING "List of specific machine types for library to target" )

if(AMDGPU_TARGETS MATCHES ";")
    set(Tensile_ARCHITECTURE "all")
else()
    set(Tensile_ARCHITECTURE ${AMDGPU_TARGETS})
endif()

# Dont build as shared as a workaround
# option( BUILD_SHARED_LIBS "Build as a shared library" ON )

list(APPEND CMAKE_MODULE_PATH ${CMAKE_CURRENT_SOURCE_DIR}/cmake)

include(virtualenv)

# Workaround broken hcc-config
macro(add_library NAME)
    if(${NAME} STREQUAL "hsa-runtime64")
        _add_library(hsa-runtime64 SHARED IMPORTED)
    else()
        _add_library(${NAME} ${ARGN})
    endif()
endmacro()
find_package(hip)

if(CMAKE_CXX_COMPILER MATCHES ".*/hcc$")
    set(TENSILE_USE_LLVM ON CACHE BOOL "Use LLVM for parsing config files.")
    set(TENSILE_USE_MSGPACK OFF CACHE BOOL "Use msgpack for parsing config files.")
    set(COMPILER "hcc")
    set(CODE_OBJECT_VERSION "V2")
    set(TENSILE_LIBRARY_FORMAT "yaml")
    set(MIOPEN_TENSILE_SRC asm_full_hcc)
    set( MIOPEN_TENSILE_TAG baed862380bf99580972f0eed51a4aab7234c2da CACHE STRING "Tensile tag to download" )
else()
    set(TENSILE_USE_LLVM OFF CACHE BOOL "Use LLVM for parsing config files.")
    set(TENSILE_USE_MSGPACK ON CACHE BOOL "Use msgpack for parsing config files.")
    set(COMPILER "hipcc")
    set(CODE_OBJECT_VERSION "V3")
    set(TENSILE_LIBRARY_FORMAT "msgpack")
    set(MIOPEN_TENSILE_SRC asm_full)
    set( MIOPEN_TENSILE_TAG 9cbabb07f81e932b9c98bf5ae48fbd7fcef615cf CACHE STRING "Tensile tag to download" )
endif()

# option(BUILD_DEV "Build for development" OFF)
# if(BUILD_DEV)
#     set(MIOPEN_TENSILE_SRC dev)
# endif()
# Use the virtual-env setup and download package from specified repo:
virtualenv_install(wheel)
virtualenv_install("git+https://github.com/ROCmSoftwarePlatform/Tensile.git@${MIOPEN_TENSILE_TAG}")
list(APPEND CMAKE_PREFIX_PATH ${VIRTUALENV_HOME_DIR})

find_package(Tensile REQUIRED COMPONENTS HIP)

set_target_properties( TensileHost PROPERTIES POSITION_INDEPENDENT_CODE ON )

get_filename_component(COMPILER_PATH ${CMAKE_CXX_COMPILER} DIRECTORY)
set(ENV{PATH} "${hip_BIN_INSTALL_DIR}:${COMPILER_PATH}:$ENV{PATH}")
TensileCreateLibraryFiles(
    "${CMAKE_CURRENT_SOURCE_DIR}/yaml/${MIOPEN_TENSILE_SRC}"
    "${CMAKE_CURRENT_BINARY_DIR}/lib/miopentensile"
    ARCHITECTURE ${Tensile_ARCHITECTURE}
    MERGE_FILES ON
    COMPILER ${COMPILER}
    CODE_OBJECT_VERSION ${CODE_OBJECT_VERSION}
    LIBRARY_FORMAT ${TENSILE_LIBRARY_FORMAT}
    VAR_PREFIX MIOPENTENSILE
    )

add_library(MIOpenTensile SHARED src/gemm_api.cpp)
add_dependencies(MIOpenTensile MIOPENTENSILE_LIBRARY_TARGET)
target_link_libraries(MIOpenTensile PUBLIC hip::host -ldl)
target_link_libraries(MIOpenTensile PRIVATE TensileHost)
target_compile_definitions(MIOpenTensile PRIVATE __HIP_PLATFORM_HCC__)

install(DIRECTORY "${CMAKE_CURRENT_BINARY_DIR}/lib/miopentensile" DESTINATION lib)

include(ROCMCreatePackage)
rocm_create_package(
    NAME MIOpenTensile
    DESCRIPTION "MIOpen and Tensile integration library"
    MAINTAINER "MIOpenTensile Support <miopen-lib.support@amd.com>"
    LDCONFIG
    DEPENDS hip-rocclr hip-base
)

rocm_install_targets(
  TARGETS MIOpenTensile
  INCLUDE
    ${CMAKE_CURRENT_SOURCE_DIR}/include
)

rocm_export_targets(
  TARGETS MIOpenTensile
  DEPENDS
    hip
)

add_subdirectory(test)