def get_swizzle_mode()

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

• 6 lines of code
• 4 McCabe index (conditional complexity)

def get_swizzle_mode(block_n: int) -> int:
    # TODO: remove some candidates if slow
    elem_size = 2
    for mode_bytes in (128, 64, 32):
        if (block_n * elem_size) % mode_bytes == 0:
            return mode_bytes
    return 0