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

[CK Tile Engine] k_block_per_cu changes in Preshuffle (#2842)

Browse Source 
* kperblock changes in CK Tile Engine Preshuffle

* Config file formatting changes
This commit is contained in:
Thrupti Raj Lakshmana Gowda
2025-09-15 15:22:11 -05:00
committed by GitHub
parent f22740df82
commit 7d7ded62d3
3 changed files with 62 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files

95
tile_engine/ops/gemm_preshuffle/configs/default_config.json
View File

@@ -1,51 +1,51 @@
{
"tile_config": {
"tile_m": {
"values": [
128
]
},
"tile_n": {
"values": [
128
]
},
"tile_k": {
"values": [
128
]
},
"warp_m": {
"values": [
1
]
},
"warp_n": {
"values": [
4
]
},
"warp_k": {
"values": [
1
]
},
"warp_tile_m": {
"values": [
16
]
},
"warp_tile_n": {
"values": [
16
]
},
"warp_tile_k": {
"values": [
16,32
]
}
"tile_config": {
"tile_m": {
"values": [
128
]
},
"tile_n": {
"values": [
128
]
},
"tile_k": {
"values": [
128
]
},
"warp_m": {
"values": [
1
]
},
"warp_n": {
"values": [
4
]
},
"warp_k": {
"values": [
1
]
},
"warp_tile_m": {
"values": [
16
]
},
"warp_tile_n": {
"values": [
16
]
},
"warp_tile_k": {
"values": [
16,32
]
}
},
"trait_config": {
"pipeline": {
"values": [
@@ -86,5 +86,6 @@
false
]
}
}
},
"k_block_per_cu": 2
}