24ef297cca3a32b6f73d14d865cee120f97674c5,torchdiffeq/_impl/solvers.py,RKAdaptiveStepsizeODESolver,_interp_fit,#RKAdaptiveStepsizeODESolver#,238

Before Change

    def _interp_fit(self, y0, y1, k, dt):
        Fit an interpolating polynomial to the results of a Runge-Kutta step.
        y_mid = y0 + k.matmul(dt * self.mid).view_as(y0).type_as(y0)  // mid is float64 so cast back
        f0 = k[..., 0]
        f1 = k[..., -1]
        return _interp_fit(y0, y1, y_mid, f0, f1, dt)

After Change

    def _interp_fit(self, y0, y1, k, dt):
        Fit an interpolating polynomial to the results of a Runge-Kutta step.
        dt = dt.type_as(y0)
        y_mid = y0 + k.matmul(dt * self.mid).view_as(y0)
        f0 = k[..., 0]
        f1 = k[..., -1]
        return _interp_fit(y0, y1, y_mid, f0, f1, dt)

Instances