fix the flatmm (#2349)

[ROCm/composable_kernel commit: d996bc78be]
This commit is contained in:
Thomas Ning
2025-06-16 02:17:53 -07:00
committed by GitHub
parent 1f77d58ae9
commit 6f158242b6
4 changed files with 8 additions and 4 deletions

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")
for s in tqdm(shapes):
total = len(shapes)
for idx, s in enumerate(shapes, 1):
run_shape_stdout_lines = run_shape(
s, profiler_bin, args["op_name"], args["dtype"], args["layout"]
)