ea0eba4488a9f84adbae44d847f17254dc6171a1,torchdiffeq/_impl/tsit5.py,Tsit5Solver,_adaptive_tsit5_step,#Tsit5Solver#Any#,110

Before Change

        if on_grid and accept_step:
            // We"ve just passed a grid point, which is typically used to indicate a discontinuity in f; we should
            // update f to match the side of the discontinuity we"re now on.
            f1 = self.func(t_next, y_next)
            if self.next_grid_index != len(self.grid_points) - 1:
                self.next_grid_index += 1
        f_next = f1 if accept_step else f0
        dt_next = _optimal_step_size(dt, mean_error_ratio, self.safety, self.ifactor, self.dfactor)

After Change

        t_next = t0 + dt + 2 * eps if accept_step else t0
        y_next = y1 if accept_step else y0
        if on_grid and accept_step:
            if eps != 0:
                // We"ve just passed a discontinuity in f; we should update f1 to match the side of the discontinuity
                // we"re now on.
                f1 = self.func(t_next.type_as(y_next[0]), y_next)
            if self.next_grid_index != len(self.grid_points) - 1:
                self.next_grid_index += 1
        f_next = f1 if accept_step else f0
        dt_next = _optimal_step_size(dt, mean_error_ratio, self.safety, self.ifactor, self.dfactor)

Instances