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

Merge commit '62a8ec155facd901232977b688d5225d72969709' into develop

Browse Source
This commit is contained in:
assistant-librarian[bot]
2025-12-24 19:11:47 +00:00
parent a169d59e06
commit 446db13a0f
4 changed files with 311 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

89
tile_engine/ops/gemm/gemm_multi_d/configs/default_ci_config.json Normal file
View File

@@ -0,0 +1,89 @@
{
"tile_config": {
"tile_m": {
"max": 256,
"min": 64,
"step": 256
},
"tile_n": {
"max": 256,
"min": 64,
"step": 256
},
"tile_k": {
"max": 256,
"min": 64,
"step": 256
},
"warp_m": {
"values": [
2
]
},
"warp_n": {
"values": [
2
]
},
"warp_k": {
"values": [
1
]
},
"warp_tile_m": {
"values": [
32
]
},
"warp_tile_n": {
"values": [
32
]
},
"warp_tile_k": {
"values": [
64
]
}
},
"trait_config": {
"pipeline": {
"values": [
"compv4"
]
},
"scheduler": {
"values": [
"intrawave",
"interwave"
]
},
"epilogue": {
"values": [
"cshuffle"
]
},
"pad_m": {
"values": [
false
]
},
"pad_n": {
"values": [
false
]
},
"pad_k": {
"values": [
false
]
},
"persistent": {
"values": [
false,
true
]
}
},
"k_block_per_cu": 1
}

89
tile_engine/ops/gemm/gemm_preshuffle/configs/default_ci_config.json Normal file
View File

@@ -0,0 +1,89 @@
{
"tile_config": {
"tile_m": {
"max": 256,
"min": 64,
"step": 256
},
"tile_n": {
"max": 256,
"min": 64,
"step": 256
},
"tile_k": {
"max": 256,
"min": 64,
"step": 256
},
"warp_m": {
"values": [
2
]
},
"warp_n": {
"values": [
2
]
},
"warp_k": {
"values": [
1
]
},
"warp_tile_m": {
"values": [
16
]
},
"warp_tile_n": {
"values": [
16
]
},
"warp_tile_k": {
"values": [
32
]
}
},
"trait_config": {
"pipeline": {
"values": [
"preshufflev2"
]
},
"scheduler": {
"values": [
"default"
]
},
"epilogue": {
"values": [
"cshuffle"
]
},
"pad_m": {
"values": [
false
]
},
"pad_n": {
"values": [
false
]
},
"pad_k": {
"values": [
false
]
},
"persistent": {
"values": [
true,
false
]
}
},
"k_block_per_cu": 1,
"permute_n": true
}

89
tile_engine/ops/gemm/gemm_universal/configs/default_ci_config.json Normal file
View File

@@ -0,0 +1,89 @@
{
"tile_config": {
"tile_m": {
"max": 256,
"min": 64,
"step": 256
},
"tile_n": {
"max": 256,
"min": 64,
"step": 256
},
"tile_k": {
"max": 256,
"min": 64,
"step": 256
},
"warp_m": {
"values": [
2
]
},
"warp_n": {
"values": [
2
]
},
"warp_k": {
"values": [
1
]
},
"warp_tile_m": {
"values": [
16
]
},
"warp_tile_n": {
"values": [
16
]
},
"warp_tile_k": {
"values": [
32
]
}
},
"trait_config": {
"pipeline": {
"values": [
"compv4"
]
},
"scheduler": {
"values": [
"intrawave",
"interwave"
]
},
"epilogue": {
"values": [
"cshuffle"
]
},
"pad_m": {
"values": [
false
]
},
"pad_n": {
"values": [
false
]
},
"pad_k": {
"values": [
false
]
},
"persistent": {
"values": [
false,
true
]
}
},
"k_block_per_cu": 1
}