Skip to content

Commit 2517b07

Browse files
committed
docs(kda): fix SM100 benchmark examples
Signed-off-by: shuyan.ycf <shuyan.ycf@antgroup.com>
1 parent f57238f commit 2517b07

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

benchmarks/bench_kda_sm100_intra_fused.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7,10 +7,10 @@
77
"""Benchmark/profiling entry for SM100 KDA K123 inverse variants.
88
99
Examples:
10-
python benchmarks/bench_kda_fwd_intra_sm100_fused_inv.py --mode both
11-
python benchmarks/bench_kda_fwd_intra_sm100_fused_inv.py --mode varlen --seq-lens 288
10+
python benchmarks/bench_kda_sm100_intra_fused.py --mode both
11+
python benchmarks/bench_kda_sm100_intra_fused.py --mode varlen --seq-lens 288
1212
/usr/local/cuda-13/bin/ncu --profile-from-start off --set full -o ncu_reports/kda_fwd_intra_sm100_varlen \
13-
.venv/bin/python benchmarks/bench_kda_fwd_intra_sm100_fused_inv.py --ncu --mode varlen
13+
.venv/bin/python benchmarks/bench_kda_sm100_intra_fused.py --ncu --mode varlen
1414
"""
1515

1616
from __future__ import annotations

0 commit comments

Comments
 (0)