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

Improve perf analysis script.

Browse Source
This commit is contained in:
Ville Pietilä
2025-09-03 13:43:40 +00:00
parent 4f8534206b
commit 2f94052cc7
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
script/perf-analysis.py
View File

@@ -66,6 +66,8 @@ def get_tflops_per_kernel(file):
def filter_by_kernel(perf_results, kernel):
"""Filter performance results by kernel name."""
if kernel == "all":
return perf_results
if kernel:
return {k: v for k, v in perf_results.items() if kernel == k.split('<')[0]}
return perf_results