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

[CK_TILE] Add pooling to ckTileEngine part4 fix suppported configurations

Browse Source
This commit is contained in:
Aleksander Dudek
2025-12-11 12:11:42 +00:00
parent 07c078d5ef
commit f6d2243288
5 changed files with 405 additions and 112 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tile_engine/ops/pooling/configs/default_config.json
View File

@@ -7,7 +7,7 @@
"values": [1]
},
"warp_m": {
"values": [1, 2]
"values": [1]
},
"warp_n": {
"values": [1]
@@ -32,4 +32,3 @@
},
"k_block_per_cu": 1
}