From fed3c417bafcee0d35b6852b8a12a04259c9315b Mon Sep 17 00:00:00 2001
From: biondizzle <biondizzle@gmail.com>
Date: Sun, 17 May 2026 09:19:10 +0000
Subject: [PATCH] Add debug OOB check for sorted_token_ids

---
 vllm/nvfp4_cutedsl.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/vllm/nvfp4_cutedsl.py b/vllm/nvfp4_cutedsl.py
index 69c1b646..33a39052 100644
--- a/vllm/nvfp4_cutedsl.py
+++ b/vllm/nvfp4_cutedsl.py
@@ -340,6 +340,9 @@ class CuTeDSLMoERunner:
         expert_offsets[1:self.num_experts + 1] = tokens_per_expert.cumsum(0)
         
         # -- Gather hidden states into slot order --
+        # DEBUG: Check for OOB indices before indexing
+        if sorted_token_ids.max() >= num_tokens or sorted_token_ids.min() < 0:
+            print(f"[CLAWMINE BUG] sorted_token_ids OOB! max={sorted_token_ids.max().item()} min={sorted_token_ids.min().item()} num_tokens={num_tokens} _token_indices_max={self._token_indices[:num_slots].max().item()}", flush=True)
         slot_hidden = hidden_states[sorted_token_ids]
         
         # === L1: gate + up ===