Test with all 61 layers (shared experts only)

2026-05-19 15:55:41 +00:00
parent 3e6041d752
commit 359654f08e
1 changed files with 2 additions and 2 deletions
--- a/tests/test_e2e_decode_b200.py
+++ b/tests/test_e2e_decode_b200.py
@@ -274,8 +274,8 @@ def main():
    runners = {}
    weights = {}
    
-    # Test with 3 representative layers: C128A, C4A, SWA
-    test_layers = [0, 2, 60]
+    # Test with all 61 layers (shared experts only)
+    test_layers = list(range(NUM_LAYERS))
    
    for layer_id in test_layers:
        p = f"model.layers.{layer_id}"