def is_tma_multicast_legal()

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

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

def is_tma_multicast_legal(shape_dim: int, block_dim: int, num_tma_multicast: int, num_sms: int,
                           require_divisible: bool = False) -> bool:
    divisible = ceil_div(shape_dim, block_dim) % num_tma_multicast == 0 or not require_divisible
    return divisible and num_sms % num_tma_multicast == 0