From 8fe5546bb33def671ee78b8644f4c8ddb488784a Mon Sep 17 00:00:00 2001 From: biondizzle Date: Tue, 19 May 2026 02:43:17 +0000 Subject: [PATCH] Fix debug script --- tests/debug_wo_a3.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/debug_wo_a3.py b/tests/debug_wo_a3.py index b276179c..8245f422 100644 --- a/tests/debug_wo_a3.py +++ b/tests/debug_wo_a3.py @@ -45,7 +45,7 @@ runner._ensure_initialized() # Compute activation gs with torch.no_grad(): - _, _, gs = quantize_to_nvfp4(o_g.reshape(T, GI)[:1]) + _, _, gs = quantize_to_nvfp4(o_g[:, 0, :]) # use first group's activation print(f"\nActivation gs from sample: {gs:.6f}") print(f"Runner gs: {runner._activation_global_scale:.6f}")