CMakeLists.txt 1.5 KB
Newer Older
Alan Turner's avatar
Alan Turner committed
1
2
3
4
5
6
7
include(Embed)
file(GLOB_RECURSE KERNEL_FILES ${CONFIGURE_DEPENDS}
    ${PROJECT_SOURCE_DIR}/include/ck/*.hpp)
message(STATUS "KERNEL_FILES: ${KERNEL_FILES}")
add_embed_library(ck_headers ${KERNEL_FILES} RELATIVE ${PROJECT_SOURCE_DIR}/build/include)

execute_process(
Paul's avatar
Paul committed
8
9
10
    COMMAND python3 ${CMAKE_CURRENT_SOURCE_DIR}/util/make_instance_strings.py
        ${PROJECT_SOURCE_DIR}/library/src/tensor_operation_instance/gpu
        ${CMAKE_CURRENT_BINARY_DIR}/solution_instances
Alan Turner's avatar
Alan Turner committed
11
12
13
    WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}/../tensor_operation_instance/gpu/
)

Paul's avatar
Paul committed
14
15
16
add_library(jit_library STATIC 
    src/device_gemm_multiple_d.cpp
    src/common.cpp
Alan Turner's avatar
Alan Turner committed
17
18
19
20
21
)
add_library(composable_kernel::jit_library ALIAS jit_library)

set_target_properties(jit_library PROPERTIES LINKER_LANGUAGE CXX)

Paul's avatar
Paul committed
22
target_include_directories(jit_library PRIVATE
Alan Turner's avatar
Alan Turner committed
23
    $<BUILD_INTERFACE:${CMAKE_CURRENT_SOURCE_DIR}/include>
Paul's avatar
Paul committed
24
    $<BUILD_INTERFACE:${PROJECT_SOURCE_DIR}/library/src/jit_library/solution_instances>
Paul's avatar
Paul committed
25
    $<BUILD_INTERFACE:${CMAKE_CURRENT_BINARY_DIR}/solution_instances>
Paul's avatar
Paul committed
26
    $<BUILD_INTERFACE:${CMAKE_CURRENT_BINARY_DIR}/embed/ck_headers/include>
Alan Turner's avatar
Alan Turner committed
27
28
29
30
31
32
33
34
35
)

target_link_libraries(jit_library PRIVATE ck_headers)

rocm_install(
    TARGETS jit_library ck_headers
    EXPORT jit_libraryTargets
)

Paul's avatar
Paul committed
36
rocm_install(DIRECTORY include/ck DESTINATION ${CMAKE_INSTALL_INCLUDEDIR})
Alan Turner's avatar
Alan Turner committed
37
38
39
40
41
42
43
44
45

rocm_install(
    EXPORT jit_libraryTargets
    FILE composable_kerneljit_libraryTargets.cmake
    NAMESPACE composable_kernel::
    DESTINATION ${CMAKE_INSTALL_LIBDIR}/cmake/composable_kernel
)