ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-17 09:08:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'develop' into zan/cK_tile/moe_gemm

Browse Source
This commit is contained in:
zanzhang
2025-05-21 14:19:00 +08:00
parent 9fe1de1971 990d645578
commit 9d8a21dfa9
376 changed files with 23855 additions and 7583 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
script/convert_miopen_driver_to_profiler.py
View File

@@ -126,6 +126,8 @@ def run_ck_grouped_conv_bwd_data(args):
args.ck_profier_op = "grouped_conv_bwd_data"
parse_data_type(args)
parse_layouts(args)
# Test all split K value from the list {1, 2, 4, 8, 32, 64, 128}
args.split_k_value = -1
cmd = [str(args.ck_profiler_cmd), str(args.ck_profier_op)]
cmd += [str(args.data_type), str(args.layout)]
@@ -136,6 +138,7 @@ def run_ck_grouped_conv_bwd_data(args):
cmd += [str(args.in_channels)]
add_conv_params_to_cmd(args, cmd)
cmd += [str(args.split_k_value)]
run_ck_profiler_cmd(cmd)