From 300f77e161efa803c5e9b5932b240e263d55cd39 Mon Sep 17 00:00:00 2001
From: Osc-7 <1449028496@qq.com>
Date: Mon, 29 Jun 2026 16:38:33 +0000
Subject: [PATCH] fix(diffusion_gemma): init preempt-rebuild runtime fields in
 init_multi_block

DiffusionGemmaReq overrides init_multi_block without calling super(), so
base DllmReq fields _resume_prefill_until and _terminal_context_block_id
were never set. lazy_activate() then raised AttributeError on the first
step(), breaking diffusion_gemma_gsm8k benchmark runs.

Fixes #45
---
 diffulex/strategy/diffusion_gemma/engine/request.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/diffulex/strategy/diffusion_gemma/engine/request.py b/diffulex/strategy/diffusion_gemma/engine/request.py
index a114123e..acf55af4 100644
--- a/diffulex/strategy/diffusion_gemma/engine/request.py
+++ b/diffulex/strategy/diffusion_gemma/engine/request.py
@@ -26,6 +26,8 @@ def init_multi_block(self, config: Config):
         self.is_multi_block = True
         self.status_history = [self.status]
         self.completion_reason = None
+        self._resume_prefill_until = 0
+        self._terminal_context_block_id: int | None = None
 
         self.block_size = config.block_size
         self.buffer_size = config.buffer_size