def _mla_softmax_reducev()

in benchmark/bench_flash_mla.py [0:0]

• 5 lines of code
• 1 McCabe index (conditional complexity)

def _mla_softmax_reducev(
    logits,
    o,
    b_seq_len,
    num_kv_splits,