From 3bd234e79c58884fa1c255e9d37ac264b6266079 Mon Sep 17 00:00:00 2001 From: Chenggang Zhao Date: Tue, 27 May 2025 13:21:19 +0800 Subject: [PATCH] Add assertions --- tests/test_core.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/test_core.py b/tests/test_core.py index ac14578..0a1b331 100644 --- a/tests/test_core.py +++ b/tests/test_core.py @@ -99,6 +99,7 @@ def construct_masked_grouped(num_groups: int, max_m: int, expected_m_per_group: masked_m = torch.empty((num_groups, ), device='cuda', dtype=torch.int) for j in range(num_groups): masked_m[j] = int(expected_m_per_group * random.uniform(0.7, 1.3)) + assert masked_m.amax().item() <= max_m return x_fp8, y_fp8, masked_m, out, ref_out