Function prefill_rows 

fn prefill_rows(
    gpu: &mut Gpu<'_>,
    rows: &[u32],
    following: Option<u32>,
    drafts: usize,
    engine: bool,
) -> Result<PrefillResume>

Advance one prompt chunk and retain the predictions needed to resume decode.