mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-06-29 11:16:59 +00:00
78 lines
2.1 KiB
CMake
78 lines
2.1 KiB
CMake
# Copyright (c) Advanced Micro Devices, Inc., or its affiliates.
|
|
# SPDX-License-Identifier: MIT
|
|
|
|
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}/..)
|
|
configure_file(${CK_ROOT}/include/ck/config.h.in ${CK_ROOT}/include/ck/config.h)
|
|
|
|
set(CMAKE_INSTALL_PREFIX "/opt/rocm" CACHE PATH "")
|
|
list(APPEND CMAKE_PREFIX_PATH /opt/rocm /opt/rocm/llvm $ENV{ROCM_PATH} $ENV{HIP_PATH})
|
|
list(APPEND CMAKE_MODULE_PATH ${CMAKE_CURRENT_SOURCE_DIR}/cmake)
|
|
|
|
if(NOT ROCmCMakeBuildTools_DIR)
|
|
set(ROCmCMakeBuildTools_DIR "/opt/rocm/lib/cmake/ROCmCMakeBuildTools")
|
|
endif()
|
|
|
|
find_package(ROCmCMakeBuildTools REQUIRED)
|
|
|
|
include(ROCMInstallTargets)
|
|
include(ROCMPackageConfigHelpers)
|
|
include(ROCMSetupVersion)
|
|
include(ROCMInstallSymlinks)
|
|
include(ROCMCreatePackage)
|
|
include(CheckCXXCompilerFlag)
|
|
include(ROCMCheckTargetIds)
|
|
include(TargetFlags)
|
|
|
|
find_package(hiprtc REQUIRED)
|
|
|
|
rocm_setup_version(VERSION 1.0)
|
|
|
|
list(APPEND CMAKE_MODULE_PATH ${CK_ROOT}/cmake)
|
|
include(Embed)
|
|
file(GLOB_RECURSE KERNEL_FILES CONFIGURE_DEPENDS
|
|
${CK_ROOT}/include/ck/*.hpp)
|
|
|
|
add_embed_library(ck_headers ${KERNEL_FILES} RELATIVE ${CK_ROOT}/include)
|
|
|
|
add_compile_options(-std=c++20)
|
|
|
|
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 hiprtc::hiprtc)
|
|
|
|
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(
|
|
TARGETS ck_host ck_headers
|
|
EXPORT ck_host_targets
|
|
INCLUDE include
|
|
)
|
|
rocm_export_targets(
|
|
TARGETS ck_host ck_headers
|
|
EXPORT ck_host_targets
|
|
NAMESPACE composable_kernel::
|
|
)
|
|
|
|
if(BUILD_TESTING)
|
|
add_subdirectory(test)
|
|
endif()
|
|
|