mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-04 05:31:24 +00:00
76ec0089fb
* Expand the base class of pool2d, prepare to share base class with pool3d * Add pool3d device op * Add pool3d f16 example * Refactor the base class. implement generic pooling in the future * clang format * get original index in max pooling * Add outputindex to base class * Fix dimension * Add pooling instance * Use indexType instead * Remove useless header * Extract IndexDataType to template * Extract pooling reference code * clang format * clang format * Fix typo * Add tensor stride * Add missing header * Add index stride and output stride * Refine naming * Add type to base class * Rename file * Use proper size * Fix typo * Refine naming * Modify the argument into vector. * Add max pool profiler * Refine naming * Support f32 pool * Fix typo * Add avg pool2d fwd in profiler * clang format * Rename AccDatatype to ComputeDatatype * Fix init * test pool * Extract variable * Add client example * Check the pooling dim * clang format * Connect argv and arg_parser * Add found check * Remove useless header * Refine naming * Adjust the order of device_pool_fwd
5 lines
294 B
CMake
5 lines
294 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_avg_pool3d_fwd avg_pool3d_fwd.cpp)
|
|
target_link_libraries(client_avg_pool3d_fwd PRIVATE composable_kernel::device_operations) |