mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-16 19:09:59 +00:00
9c0316d853
* 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
[ROCm/composable_kernel commit: 448c0f56d8]
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)
|