def _mla_softmax_reducev_kernel()

in benchmark/bench_flash_mla.py [0:0]

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

def _mla_softmax_reducev_kernel(
    Logits,
    B_seq_len,
    O,
    stride_l_b,
    stride_l_h,
    stride_l_s,
    stride_o_b,
    stride_o_h,
    NUM_KV_SPLITS: tl.constexpr,
    HEAD_DIM_CKV: tl.constexpr,