mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-05 14:11:29 +00:00
3d8d6e75e4
* Adding validation for tile sizes * Add architecture in config, and shuffle lines of code in warp_gemm.hpp * Enable MFMA for gfx950, and invalid tile handling
63 lines
1003 B
JSON
63 lines
1003 B
JSON
{
|
|
"architecture": {
|
|
"values": ["gfx90a"]
|
|
},
|
|
"layout_a": {
|
|
"values": ["r"]
|
|
},
|
|
"layout_b": {
|
|
"values": ["c"]
|
|
},
|
|
"layout_c": {
|
|
"values": ["r"]
|
|
},
|
|
"datatype": {
|
|
"values": ["fp16"]
|
|
},
|
|
"tile_m": {
|
|
"values": [256]
|
|
},
|
|
"tile_n": {
|
|
"values": [256]
|
|
},
|
|
"tile_k": {
|
|
"values": [32]
|
|
},
|
|
"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": [16]
|
|
},
|
|
"kPadM": {
|
|
"values": [false]
|
|
},
|
|
"kPadN": {
|
|
"values": [false]
|
|
},
|
|
"kPadK": {
|
|
"values": [false]
|
|
},
|
|
"pipeline": {
|
|
"values": ["compv3", "compv4", "mem"]
|
|
},
|
|
"scheduler": {
|
|
"values": ["intrawave", "interwave"]
|
|
},
|
|
"epilogue": {
|
|
"values": ["default", "cshuffle"]
|
|
}
|
|
}
|