mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-19 12:30:16 +00:00
0b07461518
* 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
[ROCm/composable_kernel commit: 866377de18]
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)
|