ywkang/kernbench2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test: fix CSV output paths in 3 sweep files (each variant now writes a unique CSV)

Browse Source
This commit is contained in:
eusonice
2026-06-26 15:14:40 -07:00
parent e20e1f7362
commit 09e70bd516
3 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/attention/test_gqa_short_context_sweep_decode_composite_fused.py
+1 -1
View File
@@ -42,7 +42,7 @@ MODES = [
CONTEXT_LENGTHS = [8 * 1024, 16 * 1024, 32 * 1024, 64 * 1024]
CSV_OUT = (Path(__file__).resolve().parents[2]
/ "docs" / "sweeps" / "short_context_decode_composite_sweep.csv")
/ "docs" / "sweeps" / "short_context_decode_composite_fused_sweep.csv")
def _compute_metrics(r, *, mode, kv_per_cube, C, S_kv):
tests/attention/test_gqa_short_context_sweep_prefill_composite.py
+1 -1
View File
@@ -42,7 +42,7 @@ MODES = [
CONTEXT_LENGTHS = [8 * 1024, 16 * 1024, 32 * 1024, 64 * 1024]
CSV_OUT = (Path(__file__).resolve().parents[2]
/ "docs" / "sweeps" / "short_context_decode_composite_sweep.csv")
/ "docs" / "sweeps" / "short_context_prefill_composite_sweep.csv")
def _compute_metrics(r, *, mode, kv_per_cube, C, S_kv):
tests/attention/test_gqa_short_context_sweep_prefill_composite_fused.py
+1 -1
View File
@@ -42,7 +42,7 @@ MODES = [
CONTEXT_LENGTHS = [8 * 1024, 16 * 1024, 32 * 1024, 64 * 1024]
CSV_OUT = (Path(__file__).resolve().parents[2]
/ "docs" / "sweeps" / "short_context_decode_composite_sweep.csv")
/ "docs" / "sweeps" / "short_context_prefill_composite_fused_sweep.csv")
def _compute_metrics(r, *, mode, kv_per_cube, C, S_kv):