mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-21 05:19:20 +00:00
e422d088a3
* 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]
17 lines
838 B
CMake
17 lines
838 B
CMake
add_custom_target(test_pool)
|
|
|
|
add_gtest_executable(test_avg_pool3d_bwd test_avg_pool3d_bwd.cpp)
|
|
add_gtest_executable(test_max_pool3d_bwd test_max_pool3d_bwd.cpp)
|
|
add_gtest_executable(test_avg_pool3d_fwd test_avg_pool3d_fwd.cpp)
|
|
add_gtest_executable(test_max_pool3d_fwd test_max_pool3d_fwd.cpp)
|
|
|
|
target_link_libraries(test_avg_pool3d_bwd PRIVATE utility device_avg_pool3d_bwd_instance)
|
|
target_link_libraries(test_max_pool3d_bwd PRIVATE utility device_max_pool_bwd_instance)
|
|
target_link_libraries(test_avg_pool3d_fwd PRIVATE utility device_pool3d_fwd_instance)
|
|
target_link_libraries(test_max_pool3d_fwd PRIVATE utility device_pool3d_fwd_instance)
|
|
|
|
add_dependencies(test_pool test_avg_pool3d_bwd)
|
|
add_dependencies(test_pool test_max_pool3d_bwd)
|
|
add_dependencies(test_pool test_avg_pool3d_fwd)
|
|
add_dependencies(test_pool test_max_pool3d_fwd)
|