def get_m_alignment_for_contiguous_layout()

in deep_gemm/jit_kernels/utils.py [0:0]

• 2 lines of code
• 1 McCabe index (conditional complexity)

def get_m_alignment_for_contiguous_layout():
    """
    When we do a grouped GEMM in contiguous format, LHS are grouped into several batches along the M axis.
    Since we deal with exactly one sub-matrix of RHS for each GEMM block, batch sizes above should align well
        with GEMM block shape.
    
    Returns:
        Group-level alignment requirement for grouped contiguous layout, which is always 128.
    """
    return 128