From 09e70bd5165fec724eb7be736796906a47cd142b Mon Sep 17 00:00:00 2001
From: eusonice <98355549+eusonice@users.noreply.github.com>
Date: Fri, 26 Jun 2026 15:14:40 -0700
Subject: [PATCH] test: fix CSV output paths in 3 sweep files (each variant now
 writes a unique CSV)

---
 .../test_gqa_short_context_sweep_decode_composite_fused.py      | 2 +-
 .../attention/test_gqa_short_context_sweep_prefill_composite.py | 2 +-
 .../test_gqa_short_context_sweep_prefill_composite_fused.py     | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/tests/attention/test_gqa_short_context_sweep_decode_composite_fused.py b/tests/attention/test_gqa_short_context_sweep_decode_composite_fused.py
index 1c8c9c0..91b1228 100644
--- a/tests/attention/test_gqa_short_context_sweep_decode_composite_fused.py
+++ b/tests/attention/test_gqa_short_context_sweep_decode_composite_fused.py
@@ -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):
diff --git a/tests/attention/test_gqa_short_context_sweep_prefill_composite.py b/tests/attention/test_gqa_short_context_sweep_prefill_composite.py
index 699af9f..73108c4 100644
--- a/tests/attention/test_gqa_short_context_sweep_prefill_composite.py
+++ b/tests/attention/test_gqa_short_context_sweep_prefill_composite.py
@@ -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):
diff --git a/tests/attention/test_gqa_short_context_sweep_prefill_composite_fused.py b/tests/attention/test_gqa_short_context_sweep_prefill_composite_fused.py
index 9a935ef..d1761df 100644
--- a/tests/attention/test_gqa_short_context_sweep_prefill_composite_fused.py
+++ b/tests/attention/test_gqa_short_context_sweep_prefill_composite_fused.py
@@ -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):