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

Code polishing x2

Browse Source
This commit is contained in:
Chenggang Zhao 2025-04-18 16:43:14 +08:00
parent a51629ddf9
commit 594020e6af
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_core.py
View File

@ -40,7 +40,7 @@ def construct(m: int, k: int, n: int) -> \
def construct_contiguous_grouped(num_groups: int, expected_m_per_group: int, k: int, n: int) -> \
Tuple[int, Tuple[torch.Tensor, torch.Tensor], Tuple[torch.Tensor, torch.Tensor], torch.Tensor, torch.Tensor]:
Tuple[int, Tuple[torch.Tensor, torch.Tensor], Tuple[torch.Tensor, torch.Tensor], torch.Tensor, torch.Tensor, torch.Tensor]:
m = 0
m_aligned = get_m_alignment_for_contiguous_layout()
group_m_list = []