Function dump_prefill_chunk 

pub(super) fn dump_prefill_chunk(
    gpu: &Gpu<'_>,
    engine: bool,
    pos: usize,
    rows: usize,
    prompt_tokens: usize,
    lines: &mut Vec<String>,
)

Record prompt argmax rows and the final long-context attention selection.