cmake_minimum_required(VERSION 3.22)
project(TMAC C CXX)

set(CMAKE_CXX_STANDARD 17)

option(TMAC_STATIC  "Build kernels_static"    OFF)
option(TMAC_LIBRARY "Build kernels library"   OFF)
include(CMakeDependentOption)
CMAKE_DEPENDENT_OPTION(TMAC_TVM "Build TVM" OFF "NOT TMAC_STATIC;NOT TMAC_LIBRARY" ON)

if(NOT DEFINED TVM_HOME)
  if(DEFINED ENV{TVM_HOME})
    set(TVM_HOME "$ENV{TVM_HOME}")
  else()
    set(TVM_HOME 3rdparty/tvm)
  endif(DEFINED ENV{TVM_HOME})
endif(NOT DEFINED TVM_HOME)
message(STATUS "TVM_HOME: ${TVM_HOME}")
if(TMAC_TVM)
  add_subdirectory(${TVM_HOME} tvm EXCLUDE_FROM_ALL)
endif()

set(
  TMAC_INCLUDE_DIRS
  ${CMAKE_CURRENT_SOURCE_DIR}/include/
  ${TVM_HOME}/include/
  ${TVM_HOME}/3rdparty/dmlc-core/include/
  ${TVM_HOME}/3rdparty/dlpack/include/
)

set(KERNEL_DEPLOY_DIR ${CMAKE_CURRENT_SOURCE_DIR}/deploy)

# add_custom_command(
#   OUTPUT ${KERNEL_DEPLOY_DIR}/tuned/kernels.o kernels.o
#   COMMAND python ${KERNEL_DEPLOY_DIR}/compile.py -t -r -o ${KERNEL_DEPLOY_DIR}/tuned -da -d intel_win -b 4 -nt 1 -tb
#   VERBATIM
# )

if(TMAC_LIBRARY)
  find_file(
    kernels_LIBRARY kernels.dll
    REQUIRED
    HINTS ${KERNEL_DEPLOY_DIR}/tuned/
  )
endif()

if(TMAC_STATIC)
  find_file(
    kernels_OBJECT kernels.o
    REQUIRED
    HINTS ${KERNEL_DEPLOY_DIR}/tuned/
  )
  add_library(kernels_objs OBJECT IMPORTED)
  set_target_properties(
    kernels_objs
    PROPERTIES
      IMPORTED_OBJECTS ${kernels_OBJECT}
  )
  add_library(kernels_static STATIC $<TARGET_OBJECTS:kernels_objs>)
  set_target_properties(kernels_static PROPERTIES LINKER_LANGUAGE CXX)
  target_link_libraries(kernels_static PUBLIC tvm_runtime)
endif()

find_file(
  tuned_kcfg kcfg.ini
  REQUIRED
  HINTS ${KERNEL_DEPLOY_DIR}/tuned/
)

find_file(
  kernels_source kernels.cc
  REQUIRED
  HINTS ${KERNEL_DEPLOY_DIR}/tuned/
)

find_file(
  kernels_header kernels.h
  REQUIRED
  HINTS ${KERNEL_DEPLOY_DIR}/tuned/
)

include(GNUInstallDirs)
set(
  TMAC_TVM_COMPILE_DEFS
  DMLC_USE_LOGGING_LIBRARY=<tvm/runtime/logging.h>
)
set(
  TMAC_LIB_INSTALL_DIR
  ${CMAKE_INSTALL_LIBDIR}
  CACHE PATH "Location of library files"
)
set(
  TMAC_INCLUDE_INSTALL_DIR
  ${CMAKE_INSTALL_INCLUDEDIR}
  CACHE PATH "Location of header files"
)

include(CMakePackageConfigHelpers)
configure_package_config_file(
  ${CMAKE_CURRENT_SOURCE_DIR}/cmake/TMACConfig.cmake.in
  ${CMAKE_CURRENT_BINARY_DIR}/TMACConfig.cmake
  INSTALL_DESTINATION ${CMAKE_INSTALL_LIBDIR}/cmake/t-mac
  PATH_VARS
    TMAC_LIB_INSTALL_DIR
    TMAC_INCLUDE_INSTALL_DIR
)

install(
  FILES ${CMAKE_CURRENT_BINARY_DIR}/TMACConfig.cmake
  DESTINATION ${CMAKE_INSTALL_LIBDIR}/cmake/t-mac
)

if(TMAC_TVM)
  install(TARGETS tvm_runtime LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR})
endif()
if(TMAC_STATIC)
  install(TARGETS kernels_static LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR})
  install(FILES ${kernels_OBJECT} DESTINATION ${CMAKE_INSTALL_LIBDIR})
endif()
if(TMAC_LIBRARY)
  install(FILES ${kernels_LIBRARY} DESTINATION ${CMAKE_INSTALL_LIBDIR})
endif()
install(DIRECTORY ${TMAC_INCLUDE_DIRS} DESTINATION ${CMAKE_INSTALL_INCLUDEDIR})
install(FILES ${kernels_header} DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}/t-mac)
install(
  FILES ${tuned_kcfg} ${kernels_source}
  DESTINATION ${CMAKE_INSTALL_LIBDIR}
)