mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-20 04:49:54 +00:00
7743be1e58
* start convnd bwd data
* add 3d laoyout name
* add conv1d reference
* add con3d reference
* finished example client code
* conv1d kernel finished
* fix input error
* add conv3d
* add 3d layout in conv_utils.hpp
* fix sepecial check
* addconvnd lib
* add test for bwd data
* finished test
* add check slice length
* convnd bwd data start
* profiler can be compiled
* fix some bug
* set input to zero
* modify readme for example
* fix test_convnd_bwd_data bug
* test_convnd_bwd_data parameter desc
* workaround for 1d
* workaroud for 2d
* change init value
* workaround for 3d int8
* fix init value bug
* remove workaround
* fix acc data type
* add int32
* change select function to template
* tilda to tilde
* remove int32 instance
* fix commit for device hpp
* fix comments for profiler
* using profile imp to test
* add pass verification
* fix conv2d reference
* fix conflict
* remove double batched_gemm
* fix exampel conv2d data and test convnd
* format
* change conv2d_bwd_data return value
* remove repeat = 1
* remove conv bwd data
Co-authored-by: ltqin <letaoqin@amd.com>
Co-authored-by: Chao Liu <chao.liu2@amd.com>
[ROCm/composable_kernel commit: 0536f2b312]
45 lines
1.8 KiB
CMake
45 lines
1.8 KiB
CMake
include_directories(BEFORE
|
|
${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
|
|
)
|
|
|
|
add_custom_target(examples)
|
|
|
|
function(add_example_executable EXAMPLE_NAME)
|
|
message("adding example ${EXAMPLE_NAME}")
|
|
add_executable(${EXAMPLE_NAME} ${ARGN})
|
|
target_link_libraries(${EXAMPLE_NAME} PRIVATE host_tensor)
|
|
add_dependencies(examples ${EXAMPLE_NAME})
|
|
endfunction(add_example_executable EXAMPLE_NAME)
|
|
|
|
add_subdirectory(01_gemm)
|
|
add_subdirectory(02_gemm_alpha_beta)
|
|
add_subdirectory(03_gemm_bias_relu)
|
|
add_subdirectory(04_gemm_bias_relu_add)
|
|
add_subdirectory(05_conv2d_fwd)
|
|
add_subdirectory(06_conv2d_fwd_bias_relu)
|
|
add_subdirectory(07_conv2d_fwd_bias_relu_add)
|
|
add_subdirectory(08_conv3d_fwd)
|
|
add_subdirectory(09_convnd_fwd)
|
|
add_subdirectory(10_conv2d_bwd_data)
|
|
add_subdirectory(11_conv2d_bwd_wgt)
|
|
add_subdirectory(12_reduce)
|
|
add_subdirectory(13_pool2d_fwd)
|
|
add_subdirectory(14_gemm_xdl_requant_relu_requant)
|
|
add_subdirectory(17_convnd_bwd_data_xdl)
|
|
add_subdirectory(15_grouped_gemm)
|
|
add_subdirectory(16_gemm_reduce)
|