mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-04 21:51:28 +00:00
4750b293fe
* General 2D Reduction Kernel * Move the reduction kernel from the example * Split the code and add the necessary policy, problem, shape files as per ck_tile convention * Add/modify the headers * Modified the example to work with the 'new' kernel * Added tests for the kernel * N-D refernce reduce * Added support for N-D input with transform to 2D * Added padding to support various input sized tensors * Bug fix in the thread buffer constructor * Some comments to explain the reduce2d block kernel * comments resolution * clang-format * comments resolution * clang-format * clang-format * comments resolution * clang-format
8 lines
210 B
CMake
8 lines
210 B
CMake
if(GPU_TARGETS MATCHES "gfx9")
|
|
add_gtest_executable(test_ck_tile_reduce2d test_reduce2d.cpp)
|
|
if(result EQUAL 0)
|
|
target_link_libraries(test_ck_tile_reduce2d PRIVATE utility)
|
|
endif()
|
|
endif()
|
|
|