@triton.jit
def masked_row(rows):
    """rows is BLOCK_M slice of the QK score
    Returns:
        BLOCK_M vector of boolean values indicating whether this
        Query x Key position is fully masked

    """
    return rows == float('-inf')
