Lines Matching refs:mctx
32 AdolcMatCtx *mctx; in PetscAdolcIJacobianVectorProduct() local
42 PetscCall(MatShellGetContext(A_shell, &mctx)); in PetscAdolcIJacobianVectorProduct()
43 m = mctx->m; in PetscAdolcIJacobianVectorProduct()
44 n = mctx->n; in PetscAdolcIJacobianVectorProduct()
47 PetscCall(TSGetDM(mctx->ts, &da)); in PetscAdolcIJacobianVectorProduct()
53 PetscCall(VecGetArrayRead(mctx->localX0, &x0)); in PetscAdolcIJacobianVectorProduct()
58 PetscCall(PetscLogEventBegin(mctx->event1, 0, 0, 0, 0)); in PetscAdolcIJacobianVectorProduct()
59 fos_forward(mctx->tag1, m, n, 0, x0, x1, NULL, action); in PetscAdolcIJacobianVectorProduct()
68 PetscCall(PetscLogEventEnd(mctx->event1, 0, 0, 0, 0)); in PetscAdolcIJacobianVectorProduct()
74 PetscCall(PetscLogEventBegin(mctx->event2, 0, 0, 0, 0)); in PetscAdolcIJacobianVectorProduct()
75 fos_forward(mctx->tag2, m, n, 0, x0, x1, NULL, action); in PetscAdolcIJacobianVectorProduct()
80 action[k] *= mctx->shift; in PetscAdolcIJacobianVectorProduct()
87 PetscCall(PetscLogEventEnd(mctx->event2, 0, 0, 0, 0)); in PetscAdolcIJacobianVectorProduct()
94 PetscCall(VecRestoreArrayRead(mctx->localX0, &x0)); in PetscAdolcIJacobianVectorProduct()
114 AdolcMatCtx *mctx; in PetscAdolcIJacobianVectorProductIDMass() local
124 PetscCall(MatShellGetContext(A_shell, &mctx)); in PetscAdolcIJacobianVectorProductIDMass()
125 m = mctx->m; in PetscAdolcIJacobianVectorProductIDMass()
126 n = mctx->n; in PetscAdolcIJacobianVectorProductIDMass()
129 PetscCall(TSGetDM(mctx->ts, &da)); in PetscAdolcIJacobianVectorProductIDMass()
135 PetscCall(VecGetArrayRead(mctx->localX0, &x0)); in PetscAdolcIJacobianVectorProductIDMass()
140 PetscCall(PetscLogEventBegin(mctx->event1, 0, 0, 0, 0)); in PetscAdolcIJacobianVectorProductIDMass()
141 fos_forward(mctx->tag1, m, n, 0, x0, x1, NULL, action); in PetscAdolcIJacobianVectorProductIDMass()
150 PetscCall(PetscLogEventEnd(mctx->event1, 0, 0, 0, 0)); in PetscAdolcIJacobianVectorProductIDMass()
158 PetscCall(VecRestoreArrayRead(mctx->localX0, &x0)); in PetscAdolcIJacobianVectorProductIDMass()
162 PetscCall(PetscLogEventBegin(mctx->event2, 0, 0, 0, 0)); in PetscAdolcIJacobianVectorProductIDMass()
163 PetscCall(VecAXPY(Y, mctx->shift, X)); in PetscAdolcIJacobianVectorProductIDMass()
164 PetscCall(PetscLogEventEnd(mctx->event2, 0, 0, 0, 0)); in PetscAdolcIJacobianVectorProductIDMass()
182 AdolcMatCtx *mctx; in PetscAdolcIJacobianTransposeVectorProduct() local
192 PetscCall(MatShellGetContext(A_shell, &mctx)); in PetscAdolcIJacobianTransposeVectorProduct()
193 m = mctx->m; in PetscAdolcIJacobianTransposeVectorProduct()
194 n = mctx->n; in PetscAdolcIJacobianTransposeVectorProduct()
197 PetscCall(TSGetDM(mctx->ts, &da)); in PetscAdolcIJacobianTransposeVectorProduct()
202 PetscCall(VecGetArrayRead(mctx->localX0, &x)); in PetscAdolcIJacobianTransposeVectorProduct()
207 PetscCall(PetscLogEventBegin(mctx->event3, 0, 0, 0, 0)); in PetscAdolcIJacobianTransposeVectorProduct()
208 if (!mctx->flg) zos_forward(mctx->tag1, m, n, 1, x, NULL); in PetscAdolcIJacobianTransposeVectorProduct()
209 fos_reverse(mctx->tag1, m, n, y, action); in PetscAdolcIJacobianTransposeVectorProduct()
218 PetscCall(PetscLogEventEnd(mctx->event3, 0, 0, 0, 0)); in PetscAdolcIJacobianTransposeVectorProduct()
224 PetscCall(PetscLogEventBegin(mctx->event4, 0, 0, 0, 0)); in PetscAdolcIJacobianTransposeVectorProduct()
225 if (!mctx->flg) { in PetscAdolcIJacobianTransposeVectorProduct()
226 zos_forward(mctx->tag2, m, n, 1, x, NULL); in PetscAdolcIJacobianTransposeVectorProduct()
227 mctx->flg = PETSC_TRUE; in PetscAdolcIJacobianTransposeVectorProduct()
229 fos_reverse(mctx->tag2, m, n, y, action); in PetscAdolcIJacobianTransposeVectorProduct()
234 action[k] *= mctx->shift; in PetscAdolcIJacobianTransposeVectorProduct()
241 PetscCall(PetscLogEventEnd(mctx->event4, 0, 0, 0, 0)); in PetscAdolcIJacobianTransposeVectorProduct()
248 PetscCall(VecRestoreArrayRead(mctx->localX0, &x)); in PetscAdolcIJacobianTransposeVectorProduct()
268 AdolcMatCtx *mctx; in PetscAdolcIJacobianTransposeVectorProductIDMass() local
278 PetscCall(MatShellGetContext(A_shell, &mctx)); in PetscAdolcIJacobianTransposeVectorProductIDMass()
279 m = mctx->m; in PetscAdolcIJacobianTransposeVectorProductIDMass()
280 n = mctx->n; in PetscAdolcIJacobianTransposeVectorProductIDMass()
283 PetscCall(TSGetDM(mctx->ts, &da)); in PetscAdolcIJacobianTransposeVectorProductIDMass()
288 PetscCall(VecGetArrayRead(mctx->localX0, &x)); in PetscAdolcIJacobianTransposeVectorProductIDMass()
293 PetscCall(PetscLogEventBegin(mctx->event3, 0, 0, 0, 0)); in PetscAdolcIJacobianTransposeVectorProductIDMass()
294 if (!mctx->flg) zos_forward(mctx->tag1, m, n, 1, x, NULL); in PetscAdolcIJacobianTransposeVectorProductIDMass()
295 fos_reverse(mctx->tag1, m, n, y, action); in PetscAdolcIJacobianTransposeVectorProductIDMass()
304 PetscCall(PetscLogEventEnd(mctx->event3, 0, 0, 0, 0)); in PetscAdolcIJacobianTransposeVectorProductIDMass()
312 PetscCall(VecRestoreArrayRead(mctx->localX0, &x)); in PetscAdolcIJacobianTransposeVectorProductIDMass()
316 PetscCall(PetscLogEventBegin(mctx->event4, 0, 0, 0, 0)); in PetscAdolcIJacobianTransposeVectorProductIDMass()
317 PetscCall(VecAXPY(X, mctx->shift, Y)); in PetscAdolcIJacobianTransposeVectorProductIDMass()
318 PetscCall(PetscLogEventEnd(mctx->event4, 0, 0, 0, 0)); in PetscAdolcIJacobianTransposeVectorProductIDMass()