From 2ddd3d0702d59170ba9bf1cba4b38b8e6748459c Mon Sep 17 00:00:00 2001 From: biondizzle Date: Tue, 19 May 2026 15:55:41 +0000 Subject: [PATCH] Test with all 61 layers (shared experts only) --- tests/test_e2e_decode_b200.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_e2e_decode_b200.py b/tests/test_e2e_decode_b200.py index 48e8a2c3..79d96e68 100644 --- 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}"