mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-12 01:10:17 +00:00
50 lines
1.4 KiB
CMake
50 lines
1.4 KiB
CMake
cmake_minimum_required(VERSION 3.16)
|
|
project(composable_kernel_host)
|
|
|
|
set(CMAKE_EXPORT_COMPILE_COMMANDS ON)
|
|
|
|
set(CMAKE_LIBRARY_OUTPUT_DIRECTORY ${CMAKE_BINARY_DIR}/lib)
|
|
set(CMAKE_ARCHIVE_OUTPUT_DIRECTORY ${CMAKE_BINARY_DIR}/lib)
|
|
set(CMAKE_RUNTIME_OUTPUT_DIRECTORY ${CMAKE_BINARY_DIR}/bin)
|
|
set(CK_ROOT ${CMAKE_CURRENT_SOURCE_DIR}/..)
|
|
|
|
find_package(ROCM)
|
|
include(ROCMInstallTargets)
|
|
include(ROCMTest)
|
|
|
|
list(APPEND CMAKE_MODULE_PATH ${CK_ROOT}/cmake)
|
|
include(Embed)
|
|
file(GLOB_RECURSE KERNEL_FILES CONFIGURE_DEPENDS
|
|
${CK_ROOT}/include/ck/*.hpp)
|
|
message(STATUS "KERNEL_FILES: ${KERNEL_FILES}")
|
|
message(STATUS "RELATIVE: ${CK_ROOT}/include")
|
|
add_embed_library(ck_headers ${KERNEL_FILES} RELATIVE ${CK_ROOT}/include)
|
|
|
|
add_definitions(-std=c++17)
|
|
|
|
file(GLOB SOURCES CONFIGURE_DEPENDS src/*.cpp)
|
|
# TODO: Use object library
|
|
add_library(ck_host STATIC ${SOURCES})
|
|
target_link_libraries(ck_host PRIVATE ck_headers)
|
|
|
|
set_target_properties(ck_host PROPERTIES
|
|
LINKER_LANGUAGE CXX
|
|
POSITION_INDEPENDENT_CODE ON)
|
|
|
|
target_include_directories(ck_host PUBLIC
|
|
$<BUILD_INTERFACE:${CMAKE_CURRENT_SOURCE_DIR}/include>
|
|
)
|
|
|
|
add_executable(ck-template-driver driver/main.cpp)
|
|
target_link_libraries(ck-template-driver ck_host)
|
|
|
|
rocm_install(
|
|
TARGETS ck_host ck_headers
|
|
EXPORT ck_hostTargets
|
|
)
|
|
rocm_install(DIRECTORY include/ck DESTINATION ${CMAKE_INSTALL_INCLUDEDIR})
|
|
|
|
if(BUILD_TESTING)
|
|
add_subdirectory(test)
|
|
endif()
|