Skip to content

Commit 0aa7572

Browse files
chenht2022claude
andcommitted
fix(epd): restore dropped bootstrap_room binding in decode generate_events
The rebase onto origin/main correctly dropped EPD's -1 bootstrap_token retry guard (subsumed by upstream's store-token-before-flipping-Success atomicity), but over-deleted the adjacent `bootstrap_room = self.receivers[req_id].bootstrap_room` binding, which is NOT part of the guard: it is the loop-local still read by pop_prefill_metadata(bootstrap_room) and pop_mrope_delta(bootstrap_room) in the same Bootstrapped+Success branch. Without it, generate_events() raises NameError on the first completed prefill->decode transfer of every EPD decode run, wedging the decode scheduler. Invisible to build / py_compile / golden tree-equality (runtime-only). Restore the single binding (matches origin/main). Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com> Signed-off-by: chenht2022 <chenht2022@gmail.com>
1 parent c9a07af commit 0aa7572

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

python/tokenspeed/runtime/pd/decode_executor.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -182,6 +182,7 @@ def generate_events(self):
182182
# message from the prefill side. bootstrap_room == bootstrap_info.bootstrap_room,
183183
# which is the key used in MooncakeKVReceiver.
184184
self._local_states[req_id] = KVPoll.Success
185+
bootstrap_room = self.receivers[req_id].bootstrap_room
185186
bootstrap_token, spec_candidate_ids = (
186187
self.kv_manager.pop_prefill_metadata(bootstrap_room)
187188
)

0 commit comments

Comments
 (0)