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

[CK_TILE] Add support for gfx12 in tile_engine for GEMM benchmarking (#2802)

Browse Source 
* initial work on adding support of gfx12 in tile_engine for GEMM benchmarking
* add stage("Run TILE_ENGINE_GEMM Tests on gfx1201") to Jenkins config
* make tile_[m/n/k] validation arch dependent
This commit is contained in:
pmaybank
2025-09-17 17:59:01 +01:00
committed by GitHub
parent c2997f2b7f
commit 592d73ad73
6 changed files with 249 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files

102
tile_engine/ops/gemm/configs/gfx120x_config.json Normal file
View File

@@ -0,0 +1,102 @@
{
"problem": {
},
"tile_config": {
"tile_m": {
"values": [
256,
128,
64
]
},
"tile_n": {
"values": [
256,
128,
64
]
},
"tile_k": {
"values": [
256,
128,
64
]
},
"warp_m": {
"values": [
4,
2,
1
]
},
"warp_n": {
"values": [
4,
2,
1
]
},
"warp_k": {
"values": [
1
]
},
"warp_tile_m": {
"values": [
16
]
},
"warp_tile_n": {
"values": [
16
]
},
"warp_tile_k": {
"values": [
16
]
}
},
"trait_config": {
"pipeline": {
"values": [
"compv3",
"mem"
]
},
"scheduler": {
"values": [
"intrawave",
"interwave"
]
},
"epilogue": {
"values": [
"cshuffle",
"default"
]
},
"pad_m": {
"values": [
false
]
},
"pad_n": {
"values": [
false
]
},
"pad_k": {
"values": [
false
]
},
"persistent": {
"values": [
false,
true
]
}
}
}