prefill optimization @hannah_chuu @medha_rv