ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-04 13:41:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

[CK_TILE] Enable persistent kernel and tail handler in tile_engine (#2300)

Browse Source 
* Enable persistent kernel in tile_engine and use tail handler

* Fix formatting

* Add persistent to default_config.json

* Remove extra newlines and add persistent also to user config

* Reduce instances from default_config.json

* add persistent to benchmark.json and custom_ci_config.json

* changed the config file to have few instances

---------

Co-authored-by: Thomas Ning <Thomas.Ning@amd.com>
Co-authored-by: ThomasNing <thomasning@amd.com>
This commit is contained in:
Sami Remes
2025-08-08 02:03:49 +03:00
committed by GitHub
parent 5d6d236b25
commit 3c9400471d
8 changed files with 60 additions and 125 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tile_engine/ops/gemm/json_config.py
View File

@@ -107,6 +107,7 @@ class TraitConfig:
pad_m: EnumConfigParam
pad_n: EnumConfigParam
pad_k: EnumConfigParam
persistent: EnumConfigParam
@dataclass
@@ -215,6 +216,9 @@ class GemmConfig:
pad_k=EnumConfigParam(
values=config_dict["trait_config"]["pad_k"]["values"]
),
persistent=EnumConfigParam(
values=config_dict["trait_config"]["persistent"]["values"]
),
)
return cls(