Test stages 1-2 first: sort + L1 GEMM

2026-05-17 21:19:23 +00:00
parent 9728604ea1
commit b05a38a9bd
1 changed files with 1 additions and 1 deletions
--- a/tests/test_pipeline_real_weights.py
+++ b/tests/test_pipeline_real_weights.py
@@ -37,7 +37,7 @@ DEVICE = "cuda"

 # Stage control (1-6)
 STAGE_START = 1
-STAGE_END = 6
+STAGE_END = 2

 # ============================================================
 # Weight loading