fix test_weight_checker_comparator assertion and ue8m0 scale unpack (#29623)

2026-06-28 23:36:37 -07:00
parent f85cc94d82
commit cd91dd0757
2 changed files with 6 additions and 1 deletions
@@ -133,7 +133,10 @@ class TestCompareQuantPair(CustomTestCase):
        reference = _compare_quant_pair(self.e_q, self.e_s, self.a_q, self.a_s)
        with patch("sglang.srt.utils.weight_checker_comparator.CHUNK_NUMEL", 128 * 128):
            chunked = _compare_quant_pair(self.e_q, self.e_s, self.a_q, self.a_s)
-        self.assertEqual(chunked, reference)
+        eq_c, max_c, mean_c, ex_c = chunked
+        eq_r, max_r, mean_r, ex_r = reference
+        self.assertEqual((eq_c, max_c, ex_c), (eq_r, max_r, ex_r))
+        self.assertAlmostEqual(mean_c, mean_r, places=7)

    @staticmethod
    def _quantize_partial(weight: torch.Tensor, scale_margin: float):