File tree Expand file tree Collapse file tree 3 files changed +9
-0
lines changed Expand file tree Collapse file tree 3 files changed +9
-0
lines changed Original file line number Diff line number Diff line change @@ -140,3 +140,6 @@ rope_bwd:
140
140
swiglu_bwd :
141
141
op : swiglu
142
142
args : --op swiglu --baseline torch_swiglu --metrics speedup --bwd --only liger_swiglu,torch_swiglu
143
+ launch_latency :
144
+ op : launch_latency
145
+ args : --op launch_latency --metrics walltime
Original file line number Diff line number Diff line change @@ -76,6 +76,8 @@ def process_manual_options(
76
76
run_configs [benchmark ]["disabled" ] = True
77
77
for benchmark in extra_args :
78
78
run_configs [benchmark ]["args" ] = extra_args [benchmark ]["args" ]
79
+ for benchmark , benchmark_config in options .get ("enabled" , {}).items ():
80
+ run_configs [benchmark ] = benchmark_config .copy ()
79
81
return run_configs
80
82
81
83
Original file line number Diff line number Diff line change @@ -7,6 +7,10 @@ disabled:
7
7
- fp8_gemm_fwd
8
8
- fp8_gemm_rowwise_fwd
9
9
- fp8_gemm_rowwise_grouped_fwd
10
+ enabled :
11
+ launch_latency :
12
+ op : launch_latency
13
+ args : --op launch_latency --metrics walltime
10
14
extra_args :
11
15
# triton_tutorial_flash_v2_opt does not work on Triton main branch
12
16
bf16_flash_attention_fwd :
You can’t perform that action at this time.
0 commit comments