upgrade input data scope to avoid cheating

Author: danielhua23

problems/amd_202602/mixed-mla/reference.py

@@ -306,13 +306,13 @@ def generate_input(batchsize: int, qseqlen: int, kvseqlen: int, seed: int) -> in
     q = torch.randn(
         (total_q, NUM_HEADS, QK_HEAD_DIM),
         dtype=torch.bfloat16, device="cuda", generator=gen,
-    ) * 0.02
+    )
 
     # Compressed KV buffer: (total_kv, 1, 576) bf16 — the source of truth
     kv_buffer_bf16 = torch.randn(
         (total_kv, NUM_KV_HEADS, QK_HEAD_DIM),
         dtype=torch.bfloat16, device="cuda", generator=gen,
-    ) * 0.02
+    )
 
     # Quantize KV to fp8
     kv_buffer_fp8, kv_scale_fp8 = quantize_fp8(kv_buffer_bf16)