ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-20 12:59:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge commit '5523df4b2dfab16d6144d7717b3b075f8c6d5104' into develop

Browse Source
This commit is contained in:
assistant-librarian[bot]
2025-06-16 15:09:06 +00:00
parent 6fc36b26e6
commit a816fbccf4
4 changed files with 4 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
script/run_ck_profiler_gemm_with_csv_shapes.py
View File

@@ -278,13 +278,13 @@ def main():
shapes = tuples(filename)
all_results = []
from tqdm import tqdm
from functools import partial
from os import path
profiler_bin = path.join(args["build_dir"], "bin", "ckProfiler")
total = len(shapes)
for idx, s in enumerate(shapes, 1):
for s in tqdm(shapes):
run_shape_stdout_lines = run_shape(
s, profiler_bin, args["op_name"], args["dtype"], args["layout"]
)