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

ckTileEngine pooling

Browse Source
This commit is contained in:
Aleksander Dudek
2026-02-10 12:50:42 +00:00
parent 3f04d27b68
commit 2c2125f73e
15 changed files with 2300 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

60
test/ck_tile/pooling_tile_engine/configs/simple_test_config.json Normal file
View File

@@ -0,0 +1,60 @@
{
"problem": {
"description": "Basic pooling functionality validation with moderate problem sizes"
},
"test_params": {
"problem_sizes_2d": [
{
"N": 1, "H": 8, "W": 8, "C": 32,
"Y": 2, "X": 2,
"stride_h": 2, "stride_w": 2,
"dilation_h": 1, "dilation_w": 1,
"pad_h_left": 0, "pad_h_right": 0,
"pad_w_left": 0, "pad_w_right": 0
},
{
"N": 2, "H": 16, "W": 16, "C": 32,
"Y": 3, "X": 3,
"stride_h": 2, "stride_w": 2,
"dilation_h": 1, "dilation_w": 1,
"pad_h_left": 1, "pad_h_right": 1,
"pad_w_left": 1, "pad_w_right": 1
},
{
"N": 1, "H": 32, "W": 32, "C": 64,
"Y": 2, "X": 2,
"stride_h": 2, "stride_w": 2,
"dilation_h": 1, "dilation_w": 1,
"pad_h_left": 0, "pad_h_right": 0,
"pad_w_left": 0, "pad_w_right": 0
}
],
"problem_sizes_3d": [
{
"N": 1, "D": 4, "H": 4, "W": 4, "C": 32,
"Z": 2, "Y": 2, "X": 2,
"stride_d": 2, "stride_h": 2, "stride_w": 2,
"dilation_d": 1, "dilation_h": 1, "dilation_w": 1,
"pad_d_left": 0, "pad_d_right": 0,
"pad_h_left": 0, "pad_h_right": 0,
"pad_w_left": 0, "pad_w_right": 0
}
]
},
"tile_config": {
"block_m": {"values": [128]},
"block_n": {"values": [1]},
"warp_m": {"values": [1]},
"warp_n": {"values": [1]},
"warp_tile_m": {"values": [128]},
"warp_tile_n": {"values": [1]},
"thread_tile_m": {"values": [2]},
"thread_tile_n": {"values": [1]}
},
"trait_config": {
"reduce_op": {"values": ["max"]},
"output_index": {"values": [true]},
"propagate_nan": {"values": [false]},
"pooling_dim": {"values": ["2d"]}
}
}