DeepGEMM/deep_gemm/jit_kernels/utils.py at 1169f83c3685613f5e156cd1984005dcedddc202 - DeepGEMM - Gitea: Git with a cup of tea

DeepSeek/DeepGEMM

mirror of https://github.com/deepseek-ai/DeepGEMM synced 2025-06-26 23:15:49 +00:00

Files

z-navy 3f92607b98 Fix get_col_major_tma_aligned_tensor to handle 2-dimensional inputs

2025-03-13 22:15:16 +08:00

3.5 KiB

Raw Blame History

View Raw