From 33e28100ee813f69e9a2de941ecbd9861d465b5a Mon Sep 17 00:00:00 2001 From: biondizzle Date: Sun, 17 May 2026 08:24:27 +0000 Subject: [PATCH] test: use runner's built-in warmup method --- tests/test_warmup_gs.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/tests/test_warmup_gs.py b/tests/test_warmup_gs.py index 6c8b2c6b..b46a7a6b 100644 --- a/tests/test_warmup_gs.py +++ b/tests/test_warmup_gs.py @@ -154,12 +154,9 @@ def main(): [w.clone() for w in l2_fp4], [w.clone() for w in l2_sf], list(l2_gs), ) - l1_gs_val, l2_gs_val = warmup_compute_gs(runner, hidden_states, topk_weights, topk_ids) - print(f" Warmup L1 gs: {l1_gs_val:.10f}") - print(f" Warmup L2 gs: {l2_gs_val:.10f}") + # Use the runner's built-in warmup method + runner.compute_activation_global_scales(hidden_states.clone(), topk_weights, topk_ids) - runner._l1_activation_global_scale = l1_gs_val - runner._l2_activation_global_scale = l2_gs_val result = runner.run(hidden_states.clone(), topk_weights, topk_ids) cos = torch.nn.functional.cosine_similarity(