ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-29 19:28:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

use values in json config, change valid warp configurations

Browse Source
This commit is contained in:
Philip Maybank
2025-09-09 15:04:20 +01:00
parent 1da4d222c6
commit 65b57a61c8
2 changed files with 10 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
tile_engine/ops/gemm/configs/gfx120x_config.json
View File

@@ -9,14 +9,16 @@
]
},
"tile_n": {
"max": 256,
"min": 64,
"step": 64
"values": [
256,
128
]
},
"tile_k": {
"max": 256,
"min": 64,
"step": 64
"values": [
256,
128
]
},
"warp_m": {
"values": [
@@ -57,7 +59,6 @@
"pipeline": {
"values": [
"compv3",
"compv4",
"mem"
]
},

3
tile_engine/ops/gemm/validation_utils.py
View File

@@ -157,7 +157,8 @@ def is_trait_combination_valid(pipeline: str, epilogue: str, scheduler: str) ->
def validate_warp_configuration(warp_m: int, warp_n: int, warp_k: int) -> bool:
"""Validate warp configuration."""
return (warp_m, warp_n, warp_k) in [(1, 4, 1), (2, 2, 1), (4, 1, 1)]
# return (warp_m, warp_n, warp_k) in [(1, 4, 1), (2, 2, 1), (4, 1, 1)]
return (warp_m, warp_n, warp_k) in [(2, 4, 1), (1, 8, 1), (8, 1, 1), (4, 2, 1)]
def validate_dimension_alignment(