ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-20 04:49:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

[ck_tile] Pooling example - Improved tile sizes (#3233)

Browse Source 
* improved tile sizes

- modified tile sizes for improved example performance

* Update example/ck_tile/36_pooling/pool3d.cpp

Co-authored-by: Adam Osewski <19374865+aosewski@users.noreply.github.com>

---------

Co-authored-by: Adam Osewski <19374865+aosewski@users.noreply.github.com>

[ROCm/composable_kernel commit: 1eb26460aa]
This commit is contained in:
Yashvardhan Agarwal
2025-11-19 16:30:18 +02:00
committed by GitHub
parent b5e2f26808
commit 94b3569da0
2 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
test/ck_tile/pooling/test_pooling.cpp
View File

@@ -295,10 +295,10 @@ class TestCkTilePooling : public ::testing::Test
}
};
using Shape1_BlockWarps = ck_tile::sequence<4, 1>;
using Shape1_BlockTile = ck_tile::sequence<128, 128>;
using Shape1_WarpTile = ck_tile::sequence<32, 128>;
using Shape1_ThreadTile = ck_tile::sequence<8, 8>;
using Shape1_BlockWarps = ck_tile::sequence<1, 1>;
using Shape1_BlockTile = ck_tile::sequence<128, 1>;
using Shape1_WarpTile = ck_tile::sequence<128, 1>;
using Shape1_ThreadTile = ck_tile::sequence<2, 1>;
// Cross-warp configuration
using Shape2_BlockWarps = ck_tile::sequence<2, 2>;