mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-12 09:16:52 +00:00
448c0f56d8
* Add pool2d instance BWD AVG * Add pool2d instance BWD MAX * Fix: avg review * Fix review: part2 * Fix - enable test when type is compiled * Fix review part3
29 lines
1.6 KiB
CMake
29 lines
1.6 KiB
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)
|
|
add_gtest_executable(test_avg_pool2d_bwd test_avg_pool2d_bwd.cpp)
|
|
add_gtest_executable(test_max_pool2d_bwd test_max_pool2d_bwd.cpp)
|
|
add_gtest_executable(test_avg_pool2d_fwd test_avg_pool2d_fwd.cpp)
|
|
add_gtest_executable(test_max_pool2d_fwd test_max_pool2d_fwd.cpp)
|
|
|
|
target_link_libraries(test_avg_pool3d_bwd PRIVATE utility device_avg_pool3d_bwd_instance)
|
|
target_link_libraries(test_avg_pool2d_bwd PRIVATE utility device_avg_pool2d_bwd_instance)
|
|
target_link_libraries(test_max_pool2d_bwd PRIVATE utility device_max_pool_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)
|
|
target_link_libraries(test_avg_pool2d_fwd PRIVATE utility device_pool2d_fwd_instance)
|
|
target_link_libraries(test_max_pool2d_fwd PRIVATE utility device_pool2d_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)
|
|
add_dependencies(test_pool test_avg_pool2d_bwd)
|
|
add_dependencies(test_pool test_max_pool2d_bwd)
|
|
add_dependencies(test_pool test_avg_pool2d_fwd)
|
|
add_dependencies(test_pool test_max_pool2d_fwd)
|