[TUTORIALS] Fix return values of the backward function (#6782)

Jokeren · whitneywhtsang · commit 7f1d43ece821 · 2025-05-13T18:30:17.000Z
diff --git a/python/tutorials/06-fused-attention.py b/python/tutorials/06-fused-attention.py
@@ -700,7 +700,7 @@ def backward(ctx, do):
             num_stages=NUM_STAGES  #
         )
 
-        return dq, dk, dv, None, None
+        return dq, dk, dv, None, None, None, None
 
 
 attention = _attention.apply

Original file line number	Diff line number	Diff line change
`@@ -700,7 +700,7 @@ def backward(ctx, do):`
`700`	`700`	`num_stages=NUM_STAGES #`
`701`	`701`	`)`
`702`	`702`
`703`		`- return dq, dk, dv, None, None`
	`703`	`+ return dq, dk, dv, None, None, None, None`
`704`	`704`
`705`	`705`
`706`	`706`	`attention = _attention.apply`