Lines Matching refs:Amat
28 PetscErrorCode TSSetRHSJacobianP(TS ts, Mat Amat, TSRHSJacobianPFn *func, PetscCtx ctx) in TSSetRHSJacobianP() argument
32 PetscValidHeaderSpecific(Amat, MAT_CLASSID, 2); in TSSetRHSJacobianP()
36 if (Amat) { in TSSetRHSJacobianP()
37 PetscCall(PetscObjectReference((PetscObject)Amat)); in TSSetRHSJacobianP()
39 ts->Jacprhs = Amat; in TSSetRHSJacobianP()
64 PetscErrorCode TSGetRHSJacobianP(TS ts, Mat *Amat, TSRHSJacobianPFn **func, PetscCtxRt ctx) in TSGetRHSJacobianP() argument
69 if (Amat) *Amat = ts->Jacprhs; in TSGetRHSJacobianP()
90 PetscErrorCode TSComputeRHSJacobianP(TS ts, PetscReal t, Vec U, Mat Amat) in TSComputeRHSJacobianP() argument
93 if (!Amat) PetscFunctionReturn(PETSC_SUCCESS); in TSComputeRHSJacobianP()
97 …llback JacobianP for sensitivity analysis", (*ts->rhsjacobianp)(ts, t, U, Amat, ts->rhsjacobianpct… in TSComputeRHSJacobianP()
100 PetscCall(MatZeroEntries(Amat)); in TSComputeRHSJacobianP()
101 PetscCall(MatAssembled(Amat, &assembled)); in TSComputeRHSJacobianP()
103 PetscCall(MatAssemblyBegin(Amat, MAT_FINAL_ASSEMBLY)); in TSComputeRHSJacobianP()
104 PetscCall(MatAssemblyEnd(Amat, MAT_FINAL_ASSEMBLY)); in TSComputeRHSJacobianP()
137 PetscErrorCode TSSetIJacobianP(TS ts, Mat Amat, PetscErrorCode (*func)(TS ts, PetscReal t, Vec U, V… in TSSetIJacobianP() argument
141 PetscValidHeaderSpecific(Amat, MAT_CLASSID, 2); in TSSetIJacobianP()
145 if (Amat) { in TSSetIJacobianP()
146 PetscCall(PetscObjectReference((PetscObject)Amat)); in TSSetIJacobianP()
148 ts->Jacp = Amat; in TSSetIJacobianP()
182 PetscErrorCode TSGetIJacobianP(TS ts, Mat *Amat, PetscErrorCode (**func)(TS ts, PetscReal t, Vec U,… in TSGetIJacobianP() argument
189 if (Amat) *Amat = ts->Jacp; in TSGetIJacobianP()
213 PetscErrorCode TSComputeIJacobianP(TS ts, PetscReal t, Vec U, Vec Udot, PetscReal shift, Mat Amat, … in TSComputeIJacobianP() argument
216 if (!Amat) PetscFunctionReturn(PETSC_SUCCESS); in TSComputeIJacobianP()
221 PetscCall(PetscLogEventBegin(TS_JacobianPEval, ts, U, Amat, 0)); in TSComputeIJacobianP()
222 …bianP for sensitivity analysis", (*ts->ijacobianp)(ts, t, U, Udot, shift, Amat, ts->ijacobianpctx)… in TSComputeIJacobianP()
226 PetscCall(MatZeroEntries(Amat)); in TSComputeIJacobianP()
227 PetscCall(MatAssembled(Amat, &assembled)); in TSComputeIJacobianP()
229 PetscCall(MatAssemblyBegin(Amat, MAT_FINAL_ASSEMBLY)); in TSComputeIJacobianP()
230 PetscCall(MatAssemblyEnd(Amat, MAT_FINAL_ASSEMBLY)); in TSComputeIJacobianP()
235 if (ts->Jacprhs == Amat) { /* No IJacobian, so we only have the RHS matrix */ in TSComputeIJacobianP()
236 PetscCall(MatScale(Amat, -1)); in TSComputeIJacobianP()
240 PetscCall(MatZeroEntries(Amat)); in TSComputeIJacobianP()
242 PetscCall(MatAXPY(Amat, -1, ts->Jacprhs, axpy)); in TSComputeIJacobianP()
245 PetscCall(PetscLogEventEnd(TS_JacobianPEval, ts, U, Amat, 0)); in TSComputeIJacobianP()
1164 PetscErrorCode TSAdjointSetRHSJacobian(TS ts, Mat Amat, PetscErrorCode (*func)(TS, PetscReal, Vec, … in TSAdjointSetRHSJacobian() argument
1168 PetscValidHeaderSpecific(Amat, MAT_CLASSID, 2); in TSAdjointSetRHSJacobian()
1172 if (Amat) { in TSAdjointSetRHSJacobian()
1173 PetscCall(PetscObjectReference((PetscObject)Amat)); in TSAdjointSetRHSJacobian()
1175 ts->Jacp = Amat; in TSAdjointSetRHSJacobian()
1186 PetscErrorCode TSAdjointComputeRHSJacobian(TS ts, PetscReal t, Vec U, Mat Amat) in TSAdjointComputeRHSJacobian() argument
1191 PetscValidHeaderSpecific(Amat, MAT_CLASSID, 4); in TSAdjointComputeRHSJacobian()
1193 …llback JacobianP for sensitivity analysis", (*ts->rhsjacobianp)(ts, t, U, Amat, ts->rhsjacobianpct… in TSAdjointComputeRHSJacobian()