mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-13 01:36:06 +00:00
866377de18
* Add maxpool instances * Rename index pool to max pool. * Add maxpool bwd bf16 instances * Add avg pool bwd instances * Rename avgpool and maxpool to avg_pool3d and max_pool * Add bf16 pool fwd instances * Add max pool bwd to ckProfiler * Add avg pool3d bwd to ckProfiler * Add avg pool bwd test * Fix bug of reference pool fwd (dilation) * Fix bug of max pool bwd (dilation and initZero) * Support bf16 compute data type * Force compute type be f32. Because atomicAdd only support f32 * Add max pool bwd test * Rename folder * Rename pool * Add max pool bwd client example * Add avg pool bwd client example * Add missing workspace * clang format * Rename macro * remove useless header * remove useless layout
12 lines
591 B
CMake
12 lines
591 B
CMake
add_executable(client_max_pool2d_fwd max_pool2d_fwd.cpp)
|
|
target_link_libraries(client_max_pool2d_fwd PRIVATE composable_kernel::device_operations)
|
|
|
|
add_executable(client_max_pool2d_bwd max_pool2d_bwd.cpp)
|
|
target_link_libraries(client_max_pool2d_bwd PRIVATE composable_kernel::device_operations)
|
|
|
|
add_executable(client_avg_pool3d_fwd avg_pool3d_fwd.cpp)
|
|
target_link_libraries(client_avg_pool3d_fwd PRIVATE composable_kernel::device_operations)
|
|
|
|
add_executable(client_avg_pool3d_bwd avg_pool3d_bwd.cpp)
|
|
target_link_libraries(client_avg_pool3d_bwd PRIVATE composable_kernel::device_operations)
|