@triton.jit
def make_bounds(offs_m, offs_n, M, N, EVEN_M: tl.constexpr, EVEN_N: tl.
    constexpr):
    if EVEN_M:
        mask = offs_n[None, :] < N
    elif EVEN_N:
        mask = offs_m[:, None] < M
    else:
        mask = (offs_m[:, None] < M) & (offs_n[None, :] < N)
    return mask
