mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-12 17:26:00 +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
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)
|