Fix overlap stream bug (#10873)

chen2016013 · phlrain · web-flow · commit b099c8896e17 · 2025-07-21T15:01:41.000+08:00
Co-authored-by: phlrain &lt;phliuhongyu@126.com&gt;
diff --git a/paddlenlp/transformers/deepseek_v2/modeling_pp.py b/paddlenlp/transformers/deepseek_v2/modeling_pp.py
@@ -1536,7 +1536,7 @@ def overlapped_forward_backward(
         ) = build_overlapped_nodes(forward_chunk, backward_chunk)
         forward_inputs = forward_pre_node.forward(forward_inputs)
         backward_input_grads = backward_pre_node.backward(backward_input_grads)
-        forward_inputs, backward_input_grads = overlap_node.forward_backward(
+        forward_inputs, backward_input_grads, _ = overlap_node.forward_backward(
             forward_inputs, backward_input_grads, event_to_wait
         )
         forward_inputs = forward_post_node.forward(forward_inputs)

Original file line number	Diff line number	Diff line change
`@@ -1536,7 +1536,7 @@ def overlapped_forward_backward(`
`1536`	`1536`	`) = build_overlapped_nodes(forward_chunk, backward_chunk)`
`1537`	`1537`	`forward_inputs = forward_pre_node.forward(forward_inputs)`
`1538`	`1538`	`backward_input_grads = backward_pre_node.backward(backward_input_grads)`
`1539`		`- forward_inputs, backward_input_grads = overlap_node.forward_backward(`
	`1539`	`+ forward_inputs, backward_input_grads, _ = overlap_node.forward_backward(`
`1540`	`1540`	`forward_inputs, backward_input_grads, event_to_wait`
`1541`	`1541`	`)`
`1542`	`1542`	`forward_inputs = forward_post_node.forward(forward_inputs)`