@@ -122,21 +122,24 @@ class cvodes_integrator_adjoint_vari : public vari_base {
122122 state_backward_(Eigen::VectorXd::Zero(N)),
123123 quad_(Eigen::VectorXd::Zero(num_args_vars)),
124124 t0_(t0),
125- nv_state_forward_(N_VMake_Serial(N, state_forward_.data(), sundials_context_)),
126- nv_state_backward_(N_VMake_Serial(N, state_backward_.data(), sundials_context_)),
127- nv_quad_(N_VMake_Serial(num_args_vars, quad_.data(), sundials_context_)),
128- nv_absolute_tolerance_forward_(
129- N_VMake_Serial (N, absolute_tolerance_forward_.data(), sundials_context_)),
130- nv_absolute_tolerance_backward_(
131- N_VMake_Serial (N, absolute_tolerance_backward_.data(), sundials_context_)),
125+ nv_state_forward_(
126+ N_VMake_Serial (N, state_forward_.data(), sundials_context_)),
127+ nv_state_backward_(
128+ N_VMake_Serial (N, state_backward_.data(), sundials_context_)),
129+ nv_quad_(
130+ N_VMake_Serial (num_args_vars, quad_.data(), sundials_context_)),
131+ nv_absolute_tolerance_forward_(N_VMake_Serial(
132+ N, absolute_tolerance_forward_.data(), sundials_context_)),
133+ nv_absolute_tolerance_backward_(N_VMake_Serial(
134+ N, absolute_tolerance_backward_.data(), sundials_context_)),
132135 A_forward_(SUNDenseMatrix(N, N, sundials_context_)),
133136 A_backward_(SUNDenseMatrix(N, N, sundials_context_)),
134- LS_forward_(
135- N == 0 ? nullptr
136- : SUNLinSol_Dense(nv_state_forward_, A_forward_, sundials_context_)),
137- LS_backward_(
138- N == 0 ? nullptr
139- : SUNLinSol_Dense(nv_state_backward_, A_backward_, sundials_context_)),
137+ LS_forward_(N == 0 ? nullptr
138+ : SUNLinSol_Dense(nv_state_forward_, A_forward_,
139+ sundials_context_)),
140+ LS_backward_(N == 0 ? nullptr
141+ : SUNLinSol_Dense(nv_state_backward_, A_backward_,
142+ sundials_context_)),
140143 cvodes_mem_(CVodeCreate(solver_forward, sundials_context_)),
141144 local_args_tuple_(deep_copy_vars(args)...),
142145 value_of_args_tuple_(value_of(args)...) {
0 commit comments