| /petsc/src/ts/impls/implicit/alpha/ |
| H A D | alpha2.c | 26 Vec A0, Aa, A1; member
81 if (th->predictor) PetscCall(th->predictor(ts, th->X0, th->V0, th->A0, X1, th->predictor_ctx)); in TSAlpha_ApplyPredictor()
109 Vec X0 = th->X0, V0 = th->V0, A0 = th->A0; in TSAlpha_StageVecs() local
120 PetscCall(VecAXPBY(A1, -(1 - 2 * Beta) / (2 * Beta), 1 / (dt * dt * Beta), A0)); in TSAlpha_StageVecs()
122 PetscCall(VecWAXPY(V1, (1.0 - Gamma) / Gamma, A0, A1)); in TSAlpha_StageVecs()
131 PetscCall(VecWAXPY(Aa, -1.0, A0, A1)); in TSAlpha_StageVecs()
132 PetscCall(VecAYPX(Aa, Alpha_m, A0)); in TSAlpha_StageVecs()
175 PetscCall(VecZeroEntries(th->A0)); in TSAlpha_Restart()
202 PetscCall(VecZeroEntries(th->A0)); in TSAlpha_Restart()
203 PetscCall(VecAXPY(th->A0, -3 / time_step, V0)); in TSAlpha_Restart()
[all …]
|
| /petsc/src/ts/tutorials/ |
| H A D | ex43.c | 53 PetscErrorCode Predictor(TS ts, Vec X0, Vec V0, Vec A0, Vec X1, PetscCtx ctx) in Predictor() argument
63 PetscCall(VecAXPY(X1, accel_fac, A0)); in Predictor()
|
| /petsc/src/mat/impls/baij/seq/ |
| H A D | baij.h | 707 …__m256d S0, S1, S2, S3, S4, S5, S6, S7, S8, B0, B1, B2, B6, B7, B8, A0, A1, A2, A3, A4, A5, A6, A… in PetscKernel_A_gets_A_times_B_9() local
721 A0 = _mm256_loadu_pd(W + 0); in PetscKernel_A_gets_A_times_B_9()
727 S0 = _mm256_fmadd_pd(A0, B0, S0); in PetscKernel_A_gets_A_times_B_9()
730 S3 = _mm256_fmadd_pd(A0, B1, S3); in PetscKernel_A_gets_A_times_B_9()
733 S6 = _mm256_fmadd_pd(A0, B2, S6); in PetscKernel_A_gets_A_times_B_9()
769 A0 = _mm256_loadu_pd(W + 27); in PetscKernel_A_gets_A_times_B_9()
775 S0 = _mm256_fmadd_pd(A0, B6, S0); in PetscKernel_A_gets_A_times_B_9()
778 S3 = _mm256_fmadd_pd(A0, B7, S3); in PetscKernel_A_gets_A_times_B_9()
781 S6 = _mm256_fmadd_pd(A0, B8, S6); in PetscKernel_A_gets_A_times_B_9()
801 A0 = _mm256_loadu_pd(W + 45); in PetscKernel_A_gets_A_times_B_9()
[all …]
|
| /petsc/src/ksp/pc/impls/hpddm/ |
| H A D | pchpddm.cxx | 2355 … Mat A0, *a; /* with an SVD: [ A_00 A_01 ] */ in PCSetUp_HPDDM() local
2412 …PetscCall(MatCreateSubMatrix(a[0], rows, cols = rows, MAT_INITIAL_MATRIX, &A0)); /* [ A_00 A_01 ;… in PCSetUp_HPDDM()
2428 …PetscCall(MatCreateSubMatrix(a[0], stride, cols, MAT_INITIAL_MATRIX, &A0)); /* [ A_00 A_01 ; A_10… in PCSetUp_HPDDM()
2432 PetscCall(MatSetOption(A0, MAT_SYMMETRIC, PETSC_TRUE)); in PCSetUp_HPDDM()
2433 PetscCall(MatConvert(A0, MATSEQSBAIJ, MAT_INPLACE_MATRIX, &A0)); in PCSetUp_HPDDM()
2440 PetscCall(ISCreateStride(PETSC_COMM_SELF, A0->rmap->n, 0, 1, &stride)); in PCSetUp_HPDDM()
2455 …PetscCall(PCHPDDMCommunicationAvoidingPCASM_Private(data->levels[0]->pc, flg ? A0 : a[0], PETSC_TR… in PCSetUp_HPDDM()
2472 PetscCall(KSPSetOperators(h->ksp, A0, A0)); in PCSetUp_HPDDM()
2522 *subA = A0; in PCSetUp_HPDDM()
2525 } else PetscCall(MatDestroy(&A0)); in PCSetUp_HPDDM()
|
| /petsc/include/ |
| H A D | petscts.h | 1577 PETSC_EXTERN_TYPEDEF typedef PetscErrorCode TSAlpha2PredictorFn(TS ts, Vec X0, Vec V0, Vec A0, Vec …
|