mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-12 09:16:52 +00:00
119712bd90
[CK_TILE] Add pooling in tile_engine ## Motivation <!-- Explain the purpose of this PR and the goals it aims to achieve. --> Add pooling in ck tile engine ## Technical Details <!-- Explain the changes along with any relevant GitHub links. --> ## Test Plan <!-- Explain any relevant testing done to verify this PR. --> ## Test Result <!-- Briefly summarize test outcomes. --> ## Submission Checklist - [ ] Look over the contributing guidelines at https://github.com/ROCm/ROCm/blob/develop/CONTRIBUTING.md#pull-requests.
21 lines
694 B
JSON
21 lines
694 B
JSON
{
|
|
"problem": {
|
|
"description": "Default pooling configuration for tile_engine benchmarks"
|
|
},
|
|
"tile_config": {
|
|
"block_m": {"values": [64,128,256]},
|
|
"block_n": {"values": [1,2]},
|
|
"warp_m": {"values": [1]},
|
|
"warp_n": {"values": [1]},
|
|
"warp_tile_m": {"values": [128]},
|
|
"warp_tile_n": {"values": [1]},
|
|
"thread_tile_m": {"values": [1,2,4]},
|
|
"thread_tile_n": {"values": [1]}
|
|
},
|
|
"trait_config": {
|
|
"reduce_op": {"values": ["max", "min", "avg"]},
|
|
"output_index": {"values": [true, false]},
|
|
"propagate_nan": {"values": [true, false]},
|
|
"pooling_dim": {"values": ["2d", "3d"]}
|
|
}
|
|
} |