import numpy as np


def online_softmax_step(m, l, acc, scores, v_block):
    raise NotImplementedError


def flash_attention(Q, K, V, block_size, causal=False):
    raise NotImplementedError