From 25ca8b9d99bdcdb34925931bd79512a1de12b684 Mon Sep 17 00:00:00 2001 From: Brian Nguyen Date: Sun, 19 Jul 2026 08:44:18 -0500 Subject: [PATCH] [https://nvbugs/6442074][fix] DSparkWorker: rename forward to _forward_impl d97397437f added __init_subclass__ to SpecWorkerBase requiring subclasses to implement _forward_impl instead of overriding forward directly, but did not update DSparkWorker. Rename the method to satisfy the contract. Signed-off-by: Brian Nguyen --- tensorrt_llm/_torch/speculative/dspark.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorrt_llm/_torch/speculative/dspark.py b/tensorrt_llm/_torch/speculative/dspark.py index f2597f5e9ef3..80425833b0be 100644 --- a/tensorrt_llm/_torch/speculative/dspark.py +++ b/tensorrt_llm/_torch/speculative/dspark.py @@ -404,7 +404,7 @@ def _draft_gen_block_batched( ) return block_logits - def forward( + def _forward_impl( self, input_ids, position_ids,