We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2fbd6ee commit 39075eaCopy full SHA for 39075ea
diffrax/_solver/reversible.py
@@ -88,7 +88,7 @@ def step(
88
y1 = (self.l * (ω(y0) - ω(z0)) + ω(step_z0)).ω
89
90
step_y1, y_error, _, _, result2 = self.solver.step(
91
- terms, t1, t0, y1, args, original_solver_state, False
+ terms, t1, t0, y1, args, original_solver_state, True
92
)
93
z1 = (ω(y1) + ω(z0) - ω(step_y1)).ω
94
@@ -113,7 +113,7 @@ def backward_step(
113
114
z0 = (ω(z1) - ω(y1) + ω(step_y1)).ω
115
step_z0, _, dense_info, _, _ = self.solver.step(
116
- terms, t0, t1, z0, args, original_solver_state, False
+ terms, t0, t1, z0, args, original_solver_state, True
117
118
y0 = ((1 / self.l) * (ω(y1) - ω(step_z0)) + ω(z0)).ω
119
solver_state = (original_solver_state, z0)
0 commit comments