From b3fdb145f8c8880d80504e1820729c2906b410fa Mon Sep 17 00:00:00 2001
From: Egon Ruiter <e.m.g.ruiter@students.uu.nl>
Date: Wed, 15 Jul 2026 20:17:45 +0200
Subject: [PATCH] feat(prefetch): add opt-in lookahead-3 prefetching option for
 PILOT=3

---
 c/olmoe.c | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/c/olmoe.c b/c/olmoe.c
index 72967df..f532a6d 100644
--- a/c/olmoe.c
+++ b/c/olmoe.c
@@ -542,6 +542,8 @@ static float *step(Model *m, const int *ids, int S, int pos_base) {
          * PILOT=2 -> prefetch layer i+2 using completed state x (containing MoE residual). */
         if (g_pilot >= 2 && S <= 8 && i + 2 < c->n_layers)
             pilot_prefetch(m, i + 2, x, S);
+        if (g_pilot >= 3 && S <= 8 && i + 3 < c->n_layers)
+            pilot_prefetch(m, i + 3, x, S);
     }
     /* IMPROVEMENT 2: count tokens; trigger hot pinning after warmup */
     m->token_count++; m->freq_token_count++;