| /petsc/src/tao/interface/ |
| H A D | taosolver_hj.c | 295 PetscErrorCode TaoComputeJacobian(Tao tao, Vec X, Mat J, Mat Jpre) in TaoComputeJacobian() argument 303 PetscCall(PetscLogEventBegin(TAO_JacobianEval, tao, X, J, Jpre)); in TaoComputeJacobian() 304 …PetscCallBack("Tao callback Jacobian", (*tao->ops->computejacobian)(tao, X, J, Jpre, tao->user_jac… in TaoComputeJacobian() 305 PetscCall(PetscLogEventEnd(TAO_JacobianEval, tao, X, J, Jpre)); in TaoComputeJacobian() 336 PetscErrorCode TaoComputeResidualJacobian(Tao tao, Vec X, Mat J, Mat Jpre) in TaoComputeResidualJacobian() argument 344 PetscCall(PetscLogEventBegin(TAO_JacobianEval, tao, X, J, Jpre)); in TaoComputeResidualJacobian() 345 …uares residual Jacobian", (*tao->ops->computeresidualjacobian)(tao, X, J, Jpre, tao->user_lsjacP)); in TaoComputeResidualJacobian() 346 PetscCall(PetscLogEventEnd(TAO_JacobianEval, tao, X, J, Jpre)); in TaoComputeResidualJacobian() 374 PetscErrorCode TaoComputeJacobianState(Tao tao, Vec X, Mat J, Mat Jpre, Mat Jinv) in TaoComputeJacobianState() argument 382 PetscCall(PetscLogEventBegin(TAO_JacobianEval, tao, X, J, Jpre)); in TaoComputeJacobianState() [all …]
|
| /petsc/src/tao/constrained/impls/admm/ftn-custom/ |
| H A D | zadmmf.c | 39 static PetscErrorCode ourtaoadmmmisfitconstraintjacobian(Tao tao, Vec x, Mat J, Mat Jpre, PetscCtx … in ourtaoadmmmisfitconstraintjacobian() argument 41 …obian, (Tao *, Vec *, Mat *, Mat *, void *, PetscErrorCode *), (&tao, &x, &J, &Jpre, _ctx, &ierr)); in ourtaoadmmmisfitconstraintjacobian() 54 static PetscErrorCode ourtaoadmmregularizerconstraintjacobian(Tao tao, Vec x, Mat J, Mat Jpre, Pets… in ourtaoadmmregularizerconstraintjacobian() argument 56 …obian, (Tao *, Vec *, Mat *, Mat *, void *, PetscErrorCode *), (&tao, &x, &J, &Jpre, _ctx, &ierr)); in ourtaoadmmregularizerconstraintjacobian() 73 PETSC_EXTERN void taoadmmsetmisfitconstraintjacobian_(Tao *tao, Mat *J, Mat *Jpre, void (*func)(Tao… in taoadmmsetmisfitconstraintjacobian_() argument 77 …if (!*ierr) *ierr = TaoADMMSetMisfitConstraintJacobian(*tao, *J, *Jpre, ourtaoadmmmisfitconstraint… in taoadmmsetmisfitconstraintjacobian_() 94 PETSC_EXTERN void taoadmmsetregularizerconstraintjacobian_(Tao *tao, Mat *J, Mat *Jpre, void (*func… in taoadmmsetregularizerconstraintjacobian_() argument 98 …if (!*ierr) *ierr = TaoADMMSetRegularizerConstraintJacobian(*tao, *J, *Jpre, ourtaoadmmregularizer… in taoadmmsetregularizerconstraintjacobian_()
|
| /petsc/src/ts/tutorials/ |
| H A D | ex17.c | 214 PetscErrorCode FormIJacobian(TS ts, PetscReal t, Vec U, Vec Udot, PetscReal a, Mat J, Mat Jpre, Pet… in FormIJacobian() argument 226 PetscCall(MatGetOwnershipRange(Jpre, &rstart, &rend)); in FormIJacobian() 254 PetscCall(MatSetValuesStencil(Jpre, 1, &row, nc, col, vals, INSERT_VALUES)); in FormIJacobian() 257 PetscCall(MatAssemblyBegin(Jpre, MAT_FINAL_ASSEMBLY)); in FormIJacobian() 258 PetscCall(MatAssemblyEnd(Jpre, MAT_FINAL_ASSEMBLY)); in FormIJacobian() 259 if (J != Jpre) { in FormIJacobian() 265 PetscCall(MatView(Jpre, PETSC_VIEWER_STDOUT_WORLD)); in FormIJacobian()
|
| H A D | ex13.c | 188 PetscErrorCode RHSJacobian(TS ts, PetscReal t, Vec U, Mat J, Mat Jpre, PetscCtx ctx) in RHSJacobian() argument 230 PetscCall(MatSetValuesStencil(Jpre, 1, &row, nc, col, val, INSERT_VALUES)); in RHSJacobian() 233 PetscCall(MatAssemblyBegin(Jpre, MAT_FINAL_ASSEMBLY)); in RHSJacobian() 234 PetscCall(MatAssemblyEnd(Jpre, MAT_FINAL_ASSEMBLY)); in RHSJacobian() 235 if (J != Jpre) { in RHSJacobian()
|
| H A D | ex34.c | 113 PetscErrorCode FormIJacobian(TS ts, PetscReal t, Vec U, Vec Udot, PetscReal a, Mat J, Mat Jpre, Pet… in FormIJacobian() argument 142 PetscCall(MatSetValuesBlocked(Jpre, 1, &row, 2, col, &vals[0][0][0], INSERT_VALUES)); in FormIJacobian() 153 PetscCall(MatSetValuesBlocked(Jpre, 1, &row, 2, col, &vals[0][0][0], INSERT_VALUES)); in FormIJacobian() 164 PetscCall(MatSetValuesBlocked(Jpre, 1, &row, 3, col, &vals[0][0][0], INSERT_VALUES)); in FormIJacobian() 170 PetscCall(MatAssemblyBegin(Jpre, MAT_FINAL_ASSEMBLY)); in FormIJacobian() 171 PetscCall(MatAssemblyEnd(Jpre, MAT_FINAL_ASSEMBLY)); in FormIJacobian() 172 if (J != Jpre) { in FormIJacobian()
|
| H A D | ex25.c | 222 PetscErrorCode FormIJacobian(TS ts, PetscReal t, Vec X, Vec Xdot, PetscReal a, Mat J, Mat Jpre, voi… in FormIJacobian() argument 248 PetscCall(MatSetValuesBlocked(Jpre, 1, &row, 1, &col, &vals[0][0], INSERT_VALUES)); in FormIJacobian() 256 PetscCall(MatSetValuesBlocked(Jpre, 1, &row, 3, col, &vals[0][0][0], INSERT_VALUES)); in FormIJacobian() 264 PetscCall(MatAssemblyBegin(Jpre, MAT_FINAL_ASSEMBLY)); in FormIJacobian() 265 PetscCall(MatAssemblyEnd(Jpre, MAT_FINAL_ASSEMBLY)); in FormIJacobian() 266 if (J != Jpre) { in FormIJacobian()
|
| H A D | ex15.c | 246 PetscErrorCode FormIJacobian(TS ts, PetscReal t, Vec U, Vec Udot, PetscReal a, Mat J, Mat Jpre, Pet… in FormIJacobian() argument 318 PetscCall(MatSetValuesStencil(Jpre, 1, &row, nc, col, vals, INSERT_VALUES)); in FormIJacobian() 321 PetscCall(MatAssemblyBegin(Jpre, MAT_FINAL_ASSEMBLY)); in FormIJacobian() 322 PetscCall(MatAssemblyEnd(Jpre, MAT_FINAL_ASSEMBLY)); in FormIJacobian() 323 if (J != Jpre) { in FormIJacobian() 329 PetscCall(PetscPrintf(PetscObjectComm((PetscObject)Jpre), "Jpre:\n")); in FormIJacobian() 330 PetscCall(MatView(Jpre, PETSC_VIEWER_STDOUT_WORLD)); in FormIJacobian()
|
| H A D | ex22.c | 229 PetscErrorCode FormIJacobian(TS ts, PetscReal t, Vec X, Vec Xdot, PetscReal a, Mat J, Mat Jpre, voi… in FormIJacobian() argument 254 PetscCall(MatSetValuesBlocked(Jpre, 1, &i, 1, &i, &v[0][0], INSERT_VALUES)); in FormIJacobian() 261 PetscCall(MatAssemblyBegin(Jpre, MAT_FINAL_ASSEMBLY)); in FormIJacobian() 262 PetscCall(MatAssemblyEnd(Jpre, MAT_FINAL_ASSEMBLY)); in FormIJacobian() 263 if (J != Jpre) { in FormIJacobian()
|
| H A D | ex35.cxx | 195 PetscErrorCode FormIJacobian(TS ts, PetscReal t, Vec X, Vec Xdot, PetscReal a, Mat J, Mat Jpre, voi… in FormIJacobian() argument 229 PetscCall(MatSetValuesBlocked(Jpre, 1, &dof, 1, &dof, &bcvals[0][0], INSERT_VALUES)); in FormIJacobian() 237 PetscCall(MatSetValuesBlocked(Jpre, 1, &row, 3, col, &vals[0][0][0], INSERT_VALUES)); in FormIJacobian() 241 PetscCall(MatAssemblyBegin(Jpre, MAT_FINAL_ASSEMBLY)); in FormIJacobian() 242 PetscCall(MatAssemblyEnd(Jpre, MAT_FINAL_ASSEMBLY)); in FormIJacobian() 243 if (J != Jpre) { in FormIJacobian()
|
| H A D | ex22f_mf.F90 | 319 subroutine FormIJacobian(ts, t, X, Xdot, shift, J, Jpre, ctx, ierr) argument 327 Mat J, Jpre 350 PetscCall(MatSetValuesBlockedLocal(Jpre, i1, [row], i1, [col], val, INSERT_VALUES, ierr)) 352 PetscCall(MatAssemblyBegin(Jpre, MAT_FINAL_ASSEMBLY, ierr)) 353 PetscCall(MatAssemblyEnd(Jpre, MAT_FINAL_ASSEMBLY, ierr)) 354 if (J /= Jpre) then 412 subroutine FormIJacobianMF(ts, t, X, Xdot, shift, J, Jpre, ctx, ierr) argument 418 Mat J, Jpre
|
| H A D | ex22f.F90 | 288 subroutine FormIJacobian(ts, t, X, Xdot, shift, J, Jpre, ctx, ierr) argument 296 Mat J, Jpre 318 PetscCall(MatSetValuesBlockedLocal(Jpre, i1, [row], i1, [col], val, INSERT_VALUES, ierr)) 320 PetscCall(MatAssemblyBegin(Jpre, MAT_FINAL_ASSEMBLY, ierr)) 321 PetscCall(MatAssemblyEnd(Jpre, MAT_FINAL_ASSEMBLY, ierr)) 322 if (J /= Jpre) then
|
| H A D | ex74.c | 144 static PetscErrorCode RHSJacobian(TS ts, PetscReal t, Vec U, Mat J, Mat Jpre, PetscCtx ctx) in RHSJacobian() argument
|
| /petsc/src/binding/petsc4py/test/ |
| H A D | test_tao.py | 137 J, Jpre, Jg = tao.getJacobianEquality() 138 Jg[0](tao, x, J, Jpre, *Jg[1], **Jg[2]) 139 self.assertTrue(J.equal(Jpre)) 178 J, Jpre, Jh = tao.getJacobianInequality() 179 Jh[0](tao, x, J, Jpre, *Jh[1], **Jh[2]) 180 self.assertTrue(J.equal(Jpre))
|
| /petsc/src/ts/tests/ |
| H A D | ex25.c | 244 PetscErrorCode FormIJacobian(TS ts, PetscReal t, Vec X, Vec Xdot, PetscReal a, Mat J, Mat Jpre, voi… in FormIJacobian() argument 270 PetscCall(MatSetValuesBlocked(Jpre, 1, &row, 1, &col, &vals[0][0], INSERT_VALUES)); in FormIJacobian() 278 PetscCall(MatSetValuesBlocked(Jpre, 1, &row, 3, col, &vals[0][0][0], INSERT_VALUES)); in FormIJacobian() 286 PetscCall(MatAssemblyBegin(Jpre, MAT_FINAL_ASSEMBLY)); in FormIJacobian() 287 PetscCall(MatAssemblyEnd(Jpre, MAT_FINAL_ASSEMBLY)); in FormIJacobian() 288 if (J != Jpre) { in FormIJacobian()
|
| /petsc/src/tao/interface/ftn-custom/ |
| H A D | ztaosolverf.c | 116 static PetscErrorCode ourtaojacobianresidualroutine(Tao tao, Vec x, Mat J, Mat Jpre, PetscCtx ctx) in ourtaojacobianresidualroutine() argument 118 …lsjac, (Tao *, Vec *, Mat *, Mat *, void *, PetscErrorCode *), (&tao, &x, &J, &Jpre, _ctx, &ierr)); in ourtaojacobianresidualroutine() 141 static PetscErrorCode ourtaojacobianinequalityroutine(Tao tao, Vec x, Mat J, Mat Jpre, PetscCtx ctx) in ourtaojacobianinequalityroutine() argument 143 …cineq, (Tao *, Vec *, Mat *, Mat *, void *, PetscErrorCode *), (&tao, &x, &J, &Jpre, _ctx, &ierr)); in ourtaojacobianinequalityroutine() 146 static PetscErrorCode ourtaojacobianequalityroutine(Tao tao, Vec x, Mat J, Mat Jpre, PetscCtx ctx) in ourtaojacobianequalityroutine() argument 148 …jaceq, (Tao *, Vec *, Mat *, Mat *, void *, PetscErrorCode *), (&tao, &x, &J, &Jpre, _ctx, &ierr)); in ourtaojacobianequalityroutine() 201 PETSC_EXTERN void taosetjacobianresidualroutine_(Tao *tao, Mat *J, Mat *Jpre, void (*func)(Tao *, V… in taosetjacobianresidualroutine_() argument 205 …if (!*ierr) *ierr = TaoSetJacobianResidualRoutine(*tao, *J, *Jpre, ourtaojacobianresidualroutine, … in taosetjacobianresidualroutine_()
|
| /petsc/src/ts/tutorials/eimex/ |
| H A D | allen_cahn.c | 148 …Code FormIJacobian(TS ts, PetscReal t, Vec U, Vec Udot, PetscReal a, Mat J, Mat Jpre, PetscCtx ctx) in FormIJacobian() argument 166 if (J != Jpre) { in FormIJacobian() 167 PetscCall(MatAssemblyBegin(Jpre, MAT_FINAL_ASSEMBLY)); in FormIJacobian() 168 PetscCall(MatAssemblyEnd(Jpre, MAT_FINAL_ASSEMBLY)); in FormIJacobian()
|
| /petsc/src/ts/impls/implicit/theta/ |
| H A D | theta.c | 278 Mat J, Jpre, quadJ = NULL, quadJp = NULL; in TSAdjointStepBEuler_Private() local 294 PetscCall(TSGetIJacobian(ts, &J, &Jpre, NULL, NULL)); in TSAdjointStepBEuler_Private() 320 PetscCall(TSComputeSNESJacobian(ts, ts->vec_sol, J, Jpre)); in TSAdjointStepBEuler_Private() 321 PetscCall(KSPSetOperators(ksp, J, Jpre)); in TSAdjointStepBEuler_Private() 363 PetscCall(TSComputeSNESJacobian(ts, ts->vec_sol, J, Jpre)); in TSAdjointStepBEuler_Private() 364 PetscCall(KSPSetOperators(ksp, J, Jpre)); in TSAdjointStepBEuler_Private() 423 Mat J, Jpre, quadJ = NULL, quadJp = NULL; in TSAdjointStep_Theta() local 436 PetscCall(TSGetIJacobian(ts, &J, &Jpre, NULL, NULL)); in TSAdjointStep_Theta() 476 PetscCall(TSComputeSNESJacobian(ts, ts->vec_sol, J, Jpre)); in TSAdjointStep_Theta() 478 PetscCall(TSComputeSNESJacobian(ts, th->X, J, Jpre)); in TSAdjointStep_Theta() [all …]
|
| /petsc/src/tao/constrained/impls/admm/ |
| H A D | admm.c | 1006 PetscErrorCode TaoADMMSetMisfitConstraintJacobian(Tao tao, Mat J, Mat Jpre, PetscErrorCode (*func)(… in TaoADMMSetMisfitConstraintJacobian() argument 1016 if (Jpre) { in TaoADMMSetMisfitConstraintJacobian() 1017 PetscValidHeaderSpecific(Jpre, MAT_CLASSID, 3); in TaoADMMSetMisfitConstraintJacobian() 1018 PetscCheckSameComm(tao, 1, Jpre, 3); in TaoADMMSetMisfitConstraintJacobian() 1028 if (Jpre) { in TaoADMMSetMisfitConstraintJacobian() 1029 PetscCall(PetscObjectReference((PetscObject)Jpre)); in TaoADMMSetMisfitConstraintJacobian() 1031 am->JApre = Jpre; in TaoADMMSetMisfitConstraintJacobian() 1052 PetscErrorCode TaoADMMSetRegularizerConstraintJacobian(Tao tao, Mat J, Mat Jpre, PetscErrorCode (*f… in TaoADMMSetRegularizerConstraintJacobian() argument 1062 if (Jpre) { in TaoADMMSetRegularizerConstraintJacobian() 1063 PetscValidHeaderSpecific(Jpre, MAT_CLASSID, 3); in TaoADMMSetRegularizerConstraintJacobian() [all …]
|
| /petsc/src/tao/constrained/impls/ipm/ |
| H A D | pdipm.c | 274 static PetscErrorCode TaoSNESJacobian_PDIPM(SNES snes, Vec X, Mat J, Mat Jpre, PetscCtx ctx) in TaoSNESJacobian_PDIPM() argument 285 PetscCall(MatGetOwnershipRanges(Jpre, &Jranges)); in TaoSNESJacobian_PDIPM() 286 PetscCall(MatGetOwnershipRange(Jpre, &Jrstart, NULL)); in TaoSNESJacobian_PDIPM() 300 PetscCall(MatSetValues(Jpre, 1, &row, 2, cols, vals, INSERT_VALUES)); in TaoSNESJacobian_PDIPM() 309 PetscCall(MatSetValues(Jpre, 1, &row, 2, cols, vals, INSERT_VALUES)); in TaoSNESJacobian_PDIPM() 324 PetscCall(MatSetValue(Jpre, row, cols[0], aa[j], INSERT_VALUES)); in TaoSNESJacobian_PDIPM() 329 PetscCall(MatSetValue(Jpre, row, row, -pdipm->deltac, INSERT_VALUES)); in TaoSNESJacobian_PDIPM() 344 PetscCall(MatSetValue(Jpre, row, cols[0], -aa[j], INSERT_VALUES)); in TaoSNESJacobian_PDIPM() 349 PetscCall(MatSetValue(Jpre, row, row, -pdipm->deltac, INSERT_VALUES)); in TaoSNESJacobian_PDIPM() 378 PetscCall(MatSetValue(Jpre, row, cols[0], aa[j] + pdipm->deltaw, INSERT_VALUES)); in TaoSNESJacobian_PDIPM() [all …]
|
| /petsc/src/ts/tutorials/power_grid/ |
| H A D | ex8.c | 275 PetscErrorCode IJacobian(TS ts, PetscReal t, Vec X, Vec Xdot, PetscReal a, Mat J, Mat Jpre, PetscCt… in IJacobian() argument 323 PetscCall(MatSetValuesStencil(Jpre, 1, &row, nc, col, val, INSERT_VALUES)); in IJacobian() 328 PetscCall(MatAssemblyBegin(Jpre, MAT_FINAL_ASSEMBLY)); in IJacobian() 329 PetscCall(MatAssemblyEnd(Jpre, MAT_FINAL_ASSEMBLY)); in IJacobian() 330 if (J != Jpre) { in IJacobian()
|
| H A D | ex7.c | 299 PetscErrorCode IJacobian(TS ts, PetscReal t, Vec X, Vec Xdot, PetscReal a, Mat J, Mat Jpre, PetscCt… in IJacobian() argument 347 PetscCall(MatSetValuesStencil(Jpre, 1, &row, nc, col, val, INSERT_VALUES)); in IJacobian() 352 PetscCall(MatAssemblyBegin(Jpre, MAT_FINAL_ASSEMBLY)); in IJacobian() 353 PetscCall(MatAssemblyEnd(Jpre, MAT_FINAL_ASSEMBLY)); in IJacobian() 354 if (J != Jpre) { in IJacobian()
|
| H A D | ex6.c | 304 PetscErrorCode IJacobian(TS ts, PetscReal t, Vec X, Vec Xdot, PetscReal a, Mat J, Mat Jpre, PetscCt… in IJacobian() argument 453 PetscCall(MatSetValuesStencil(Jpre, 1, &row, nc, col, val, INSERT_VALUES)); in IJacobian() 458 PetscCall(MatAssemblyBegin(Jpre, MAT_FINAL_ASSEMBLY)); in IJacobian() 459 PetscCall(MatAssemblyEnd(Jpre, MAT_FINAL_ASSEMBLY)); in IJacobian() 460 if (J != Jpre) { in IJacobian()
|
| /petsc/src/ksp/ksp/interface/ |
| H A D | itfunc.c | 1106 Mat J, Jpre; in KSPUseExplicitTranspose_Private() local 1109 PetscCall(KSPGetOperators(ksp, &J, &Jpre)); in KSPUseExplicitTranspose_Private() 1112 if (J != Jpre) PetscCall(MatTranspose(Jpre, MAT_INITIAL_MATRIX, &ksp->transpose.BT)); in KSPUseExplicitTranspose_Private() 1116 if (J != Jpre) PetscCall(MatTranspose(Jpre, MAT_REUSE_MATRIX, &ksp->transpose.BT)); in KSPUseExplicitTranspose_Private() 1118 if (J == Jpre && ksp->transpose.BT != ksp->transpose.AT) { in KSPUseExplicitTranspose_Private() 1154 Mat J, Jpre; in KSPSolveTranspose() local 1155 PetscCall(KSPGetOperators(ksp, &J, &Jpre)); in KSPSolveTranspose() 1158 if (J != Jpre) PetscCall(MatTranspose(Jpre, MAT_INITIAL_MATRIX, &ksp->transpose.BT)); in KSPSolveTranspose() 1162 if (J != Jpre) PetscCall(MatTranspose(Jpre, MAT_REUSE_MATRIX, &ksp->transpose.BT)); in KSPSolveTranspose() 1164 if (J == Jpre && ksp->transpose.BT != ksp->transpose.AT) { in KSPSolveTranspose()
|
| /petsc/src/snes/tutorials/network/power/ |
| H A D | power2.c | 158 PetscErrorCode FormJacobian_Subnet(DM networkdm, Vec localX, Mat J, Mat Jpre, PetscInt nv, PetscInt… in FormJacobian_Subnet() argument 339 PetscErrorCode FormJacobian(SNES snes, Vec X, Mat J, Mat Jpre, void *appctx) in FormJacobian() argument 357 PetscCall(FormJacobian_Subnet(networkdm, localX, J, Jpre, nv, ne, vtx, edges, appctx)); in FormJacobian() 361 PetscCall(FormJacobian_Subnet(networkdm, localX, J, Jpre, nv, ne, vtx, edges, appctx)); in FormJacobian()
|
| /petsc/src/tao/leastsquares/tutorials/ |
| H A D | tomography.c | 149 PetscErrorCode EvaluateJacobian(Tao tao, Vec X, Mat J, Mat Jpre, void *ptr) in EvaluateJacobian() argument
|