From b05a38a9bdb236cb15a7776850d884d028776d3d Mon Sep 17 00:00:00 2001
From: biondizzle <biondizzle@gmail.com>
Date: Sun, 17 May 2026 21:19:23 +0000
Subject: [PATCH] Test stages 1-2 first: sort + L1 GEMM

---
 tests/test_pipeline_real_weights.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_pipeline_real_weights.py b/tests/test_pipeline_real_weights.py
index 2e2f2573..570c6ac0 100644
--- 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