[None] Fix warning when capturing CUDA graph (#9746)

Signed-off-by: ziyixiong-nv <[email protected]>

Author: ziyixiong-nv

tensorrt_llm/_torch/speculative/drafting_loops.py

@@ -19,6 +19,9 @@
 from tensorrt_llm._torch.speculative.interface import SpecMetadata
 from tensorrt_llm._torch.speculative.spec_tree_manager import SpecTreeManager
 
+# Enable capture_scalar_outputs to avoid graph breaks from Tensor.item() calls
+torch._dynamo.config.capture_scalar_outputs = True
+
 
 class BaseDraftingLoopWrapper(ABC, torch.nn.Module):