mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-17 11:30:02 +00:00
Fix the CK Tile related operators (#2356)
* fix the flatmm
* Fix the pipeline
* address the comment
[ROCm/composable_kernel commit: 3c4cdfac4f]
This commit is contained in:
@@ -278,13 +278,17 @@ 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):
|
||||
try:
|
||||
from tqdm import tqdm as iterate
|
||||
except ImportError:
|
||||
iterate = lambda x: x
|
||||
|
||||
for s in iterate(shapes):
|
||||
run_shape_stdout_lines = run_shape(
|
||||
s, profiler_bin, args["op_name"], args["dtype"], args["layout"]
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user