Short name describing what triggered the graph break
autograd.Function.apply: mark_dirty input not returned
Values or code snippet captured at the break point
Explanation of why the graph break was triggered
Dynamo only supports ctx.mark_dirty when each dirty input is also returned from forward.
Hints on how to resolve the graph break
torch.compile call, or by using torch.compiler.set_stance("force_eager").