build

CMakeLists.txt

@@ -71,17 +71,18 @@ if( DEFINED CK_OVERRIDE_HIP_VERSION_PATCH )
 endif()
 message(STATUS "Build with HIP ${HIP_VERSION}")
 
-## half
-#find_path(HALF_INCLUDE_DIR half.hpp)
-set(HALF_INCLUDE_DIR "${PROJECT_SOURCE_DIR}/external/half/include")
-message("HALF_INCLUDE_DIR: ${HALF_INCLUDE_DIR}")
+### half
+##find_path(HALF_INCLUDE_DIR half.hpp)
+#set(HALF_INCLUDE_DIR "${PROJECT_SOURCE_DIR}/external/half/include")
+#message("HALF_INCLUDE_DIR: ${HALF_INCLUDE_DIR}")
 
 
 rocm_create_package(
     NAME CK-${CK_BACKEND}
-    DESCRIPTION "High Performance Composable Kernels for AMD GPUs"
+    DESCRIPTION "High Performance Composable Kernel for AMD GPUs"
     LDCONFIG
 )
+
 ## tidy
 include(EnableCompilerWarnings)
 set(CK_TIDY_ERRORS ERRORS * -readability-inconsistent-declaration-parameter-name)
@@ -214,58 +215,31 @@ enable_cppcheck(
         unmatchedSuppression
     FORCE
     SOURCES
-        host/host_tensor/src
-        host/driver_offline/src
-        composable_kernel/src/kernel_wrapper
+        library/src
     INCLUDE
-        host/host_tensor/include
-        host/device/include
-        host/solver/include
-        host/driver_offline/include
-        composable_kernel/include/*
         ${CMAKE_CURRENT_SOURCE_DIR}/include
         ${CMAKE_CURRENT_BINARY_DIR}/include
+        ${CMAKE_CURRENT_SOURCE_DIR}/library/include
     DEFINE
         CPPCHECK=1
         __linux__=1
 )
+
 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)
 
-file(GLOB_RECURSE COMPOSABLE_KERNEL_HEADERS "composable_kernel/include/*/*.hpp")
-file(GLOB_RECURSE DEVICE_OPS_HEADERS "device_operation/include/*.hpp")
-
-file(GLOB_RECURSE DEVICE_OPS_SOURCE "device_operation/*.cpp")
-
-set(CK_HEADERS ${COMPOSABLE_KERNEL_HEADERS} ${DEVICE_OPS_HEADERS})
-set(CK_SOURCE ${DEVICE_OPS_SOURCE})
-add_library(composable_kernel ${CK_SOURCE})
+#file(GLOB_RECURSE COMPOSABLE_KERNEL_HEADERS "composable_kernel/include/*/*.hpp")
+#file(GLOB_RECURSE DEVICE_OPS_HEADERS "device_operation/include/*.hpp")
+#
+#file(GLOB_RECURSE DEVICE_OPS_SOURCE "device_operation/*.cpp")
+#
+#set(CK_HEADERS ${COMPOSABLE_KERNEL_HEADERS} ${DEVICE_OPS_HEADERS})
+#set(CK_SOURCE ${DEVICE_OPS_SOURCE})
+#add_library(composable_kernel ${CK_SOURCE})
 
+configure_file("${PROJECT_SOURCE_DIR}/include/ck/hip_version.hpp.in" "${PROJECT_BINARY_DIR}/include/ck/hip_version.hpp")
 
-target_include_directories(composable_kernel PUBLIC
-    $<BUILD_INTERFACE:${PROJECT_SOURCE_DIR}/composable_kernel/include>
-)
-target_include_directories(composable_kernel PUBLIC 
-    $<BUILD_INTERFACE:${PROJECT_SOURCE_DIR}/device_operation/include>
-)
-target_include_directories(composable_kernel PUBLIC 
-    $<BUILD_INTERFACE:${PROJECT_SOURCE_DIR}/host/include>
-)
-target_include_directories(composable_kernel PUBLIC 
-    $<BUILD_INTERFACE:${PROJECT_SOURCE_DIR}/host/host_tensor/include>
-)
-# The following should eventually be removed
-target_include_directories(composable_kernel PUBLIC
-    $<BUILD_INTERFACE:${PROJECT_SOURCE_DIR}/composable_kernel/include/utility>
-)
-target_include_directories(composable_kernel PUBLIC
-    $<BUILD_INTERFACE:${PROJECT_SOURCE_DIR}/composable_kernel/include/tensor_operation>
-)
-target_include_directories(composable_kernel PUBLIC
-    $<BUILD_INTERFACE:${PROJECT_SOURCE_DIR}/composable_kernel/include/tensor_description>
-)
-# clang_tidy_check(composable_kernel)
 SET(BUILD_DEV ON CACHE BOOL "BUILD_DEV")
 if(BUILD_DEV)
     target_compile_options(composable_kernel PRIVATE -Werror)
@@ -273,10 +247,7 @@ if(BUILD_DEV)
 endif()
 message("CMAKE_CXX_FLAGS: ${CMAKE_CXX_FLAGS}")
 
-configure_file("${PROJECT_SOURCE_DIR}/composable_kernel/include/hip_version.hpp.in" "${PROJECT_BINARY_DIR}/composable_kernel/include/hip_version.hpp")
-
-add_subdirectory(host)
-add_subdirectory(device_operation)
+add_subdirectory(library)
 add_subdirectory(example)
-add_subdirectory(profiler)
 add_subdirectory(test)
+add_subdirectory(profiler)

example/CMakeLists.txt

 include_directories(BEFORE
-    ${PROJECT_SOURCE_DIR}
-    ${PROJECT_SOURCE_DIR}/host/host_tensor/include
-    ${PROJECT_SOURCE_DIR}/host/device/include
-    ${PROJECT_SOURCE_DIR}/device_operation/include
-    ${PROJECT_SOURCE_DIR}/reference_operation/include
-    ${PROJECT_SOURCE_DIR}/composable_kernel/include
-    ${PROJECT_SOURCE_DIR}/composable_kernel/include/utility
-    ${PROJECT_SOURCE_DIR}/composable_kernel/include/tensor_description
-    ${PROJECT_SOURCE_DIR}/composable_kernel/include/tensor_operation
-    ${PROJECT_SOURCE_DIR}/composable_kernel/include/problem_transform
-    ${PROJECT_SOURCE_DIR}/external/rocm/include
-    ${PROJECT_SOURCE_DIR}/device_operation_reference/include
+    ${PROJECT_SOURCE_DIR}/include/ck
+    ${PROJECT_SOURCE_DIR}/include/ck/utility
+    ${PROJECT_SOURCE_DIR}/include/ck/tensor_description
+    ${PROJECT_SOURCE_DIR}/include/ck/tensor
+    ${PROJECT_SOURCE_DIR}/include/ck/problem_transform
+    ${PROJECT_SOURCE_DIR}/include/ck/tensor_operation/gpu/device
+    ${PROJECT_SOURCE_DIR}/include/ck/tensor_operation/gpu/grid
+    ${PROJECT_SOURCE_DIR}/include/ck/tensor_operation/gpu/block
+    ${PROJECT_SOURCE_DIR}/include/ck/tensor_operation/gpu/warp
+    ${PROJECT_SOURCE_DIR}/include/ck/tensor_operation/gpu/thread
+    ${PROJECT_SOURCE_DIR}/include/ck/tensor_operation/gpu/element
+    ${PROJECT_SOURCE_DIR}/library/include/ck/library/host_tensor
+    ${PROJECT_SOURCE_DIR}/library/include/ck/library/reference_tensor_operation/cpu
+    ${PROJECT_SOURCE_DIR}/library/include/ck/library/reference_tensor_operation/gpu
+    ${PROJECT_SOURCE_DIR}/external/include/half
 )
 
 set(GEMM_XDL_SOURCE 1_gemm_xdl/gemm_xdl.cpp)
@@ -66,4 +69,3 @@ target_link_libraries(convnd_fwd_xdl PRIVATE host_tensor)
 target_link_libraries(conv2d_bwd_data_xdl PRIVATE host_tensor)
 target_link_libraries(pool2d_fwd PRIVATE host_tensor)
 target_link_libraries(reduce_blockwise PRIVATE host_tensor)
-