DeepSeek/DeepGEMM
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepGEMM synced 2025-06-26 23:15:49 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add assertions

Browse Source
This commit is contained in:
Chenggang Zhao 2025-05-27 13:21:19 +08:00
parent 780b4098e4
commit 3bd234e79c
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/test_core.py
View File

@ -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