mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-01 12:11:19 +00:00
7b6451b68e
* Pooling 2D/3D with refernce * Tests & cleanup - added test for ppoling - cleanup - removed 2d example * Comment resolution - README added - example target name rectified - appropriate arg description and comments added * clang-format * appropriate blocksize calc * modifications for future indexing addition - instead of transforming views we now transform the descriptors, so that the same descriptor can be re-used for index tensor in the future * some basic fixes * comment resolutions * comment resolutions --------- Co-authored-by: Illia Silin <98187287+illsilin@users.noreply.github.com>
43 lines
1.7 KiB
Markdown
43 lines
1.7 KiB
Markdown
# Pooling Operator
|
|
|
|
This folder contains example for the pooling operator using ck_tile tile-programming implementation. Currently the pooling kernel only supports 2D and 3D pooling.
|
|
|
|
## build
|
|
```
|
|
# in the root of ck_tile
|
|
mkdir build && cd build
|
|
# you can replace <arch> with the appropriate architecture (for example gfx90a or gfx942) or leave it blank
|
|
../script/cmake-ck-dev.sh ../ <arch>
|
|
# The 3D pooling example
|
|
make tile_example_pool3d -j`nproc`
|
|
```
|
|
This will result in an executable `build/bin/tile_example_pool3d`
|
|
|
|
## example
|
|
```
|
|
args:
|
|
-N batch size (default:2)
|
|
-D depth dimension (default:30)
|
|
-H height dimension (default:30)
|
|
-W width dimension (default:30)
|
|
-C channel dimension (default:32)
|
|
-Z pooling window depth (default:2)
|
|
-Y pooling window height (default:2)
|
|
-X pooling window width (default:2)
|
|
-Sz window stride depth (default:2)
|
|
-Sy window stride height (default:2)
|
|
-Sx window stride width (default:2)
|
|
-Dz window dilation depth (default:1)
|
|
-Dy window dilation height (default:1)
|
|
-Dx window dilation width (default:1)
|
|
-LeftPz left padding depth (default:1)
|
|
-LeftPy left padding height (default:1)
|
|
-LeftPx left padding width (default:1)
|
|
-RightPz right padding depth (default:1)
|
|
-RightPy right padding height (default:1)
|
|
-RightPx right padding width (default:1)
|
|
-v 0: No validation, 1: CPU validation (default:1)
|
|
-warmup number of iterations before benchmark (default:0)
|
|
-repeat number of iterations to benchmark (default:1)
|
|
```
|