mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-01 12:11:19 +00:00
8b185e872e
* Partial Progress : Preshuffle working code for datatype * Partial Progress : Preshuffle Cleanup * Working code for default config with min max step * Partial Progress : PermuteN implemented in validation * Partial Progress : PermuteN changes in Preshuffle * CK Tile Engine Preshuffle Complete * CK TILE ENGINE : Preshuffle Layout validation * CK Tile Engine Preshuffle Validation * Preshuffle Validation check * CK Tile Engine Preshuffle : Fixing Validation Cases * Addressing PR review Comments * Changes in config * Addressing Review Comments * Adding additional architecture in Jenkins * Partial Progress : Selective Datatype and layouts * Limited datatypes and layouts * Addressing CI errors * Datatype updates * Datatype updates * Datatype changes to Preshuffle * Addressing Review Comments * Addressing Review Comments * Datatype changes * Changes to Cmake * Update on Jenkins * Formatting with precommit * Ruff Formatting
102 lines
1.3 KiB
JSON
102 lines
1.3 KiB
JSON
{
|
|
"tile_config": {
|
|
"tile_m": {
|
|
"max": 256,
|
|
"min": 64,
|
|
"step": 64
|
|
},
|
|
"tile_n": {
|
|
"max": 256,
|
|
"min": 64,
|
|
"step": 64
|
|
},
|
|
"tile_k": {
|
|
"max": 256,
|
|
"min": 64,
|
|
"step": 64
|
|
},
|
|
"warp_m": {
|
|
"values": [
|
|
4,
|
|
2,
|
|
1
|
|
]
|
|
},
|
|
"warp_n": {
|
|
"values": [
|
|
4,
|
|
2,
|
|
1
|
|
]
|
|
},
|
|
"warp_k": {
|
|
"values": [
|
|
1
|
|
]
|
|
},
|
|
"warp_tile_m": {
|
|
"values": [
|
|
4,
|
|
16,
|
|
32
|
|
]
|
|
},
|
|
"warp_tile_n": {
|
|
"values": [
|
|
16,
|
|
32,
|
|
64
|
|
]
|
|
},
|
|
"warp_tile_k": {
|
|
"values": [
|
|
8,
|
|
16,
|
|
32,
|
|
64,
|
|
128
|
|
]
|
|
}
|
|
},
|
|
"trait_config": {
|
|
"pipeline": {
|
|
"values": [
|
|
"preshufflev2"
|
|
]
|
|
},
|
|
"scheduler": {
|
|
"values": [
|
|
"default"
|
|
]
|
|
},
|
|
"epilogue": {
|
|
"values": [
|
|
"default",
|
|
"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
|
|
} |