| /petsc/src/ksp/ksp/impls/gmres/ |
| H A D | gmreig.c | 120 PetscScalar *wr, *wi; /* Real and imaginary part of the Ritz values */ in KSPComputeRitz_GMRES() local
140 PetscCall(PetscMalloc4(bN * bN, &H, bn * bn, &Q, bn, &wr, bn, &wi)); in KSPComputeRitz_GMRES()
187 …PetscCallBLAS("LAPACKgeev", LAPACKgeev_("N", "V", &bn, H, &bN, wr, wi, &sdummy, &idummy, Q, &bn, w… in KSPComputeRitz_GMRES()
190 …PetscCallBLAS("LAPACKgeev", LAPACKgeev_("N", "V", &bn, H, &bN, wr, &sdummy, &idummy, Q, &bn, work,… in KSPComputeRitz_GMRES()
200 for (i = 0; i < bn; i++) modul[i] = PetscAbsScalar(wr[i]); in KSPComputeRitz_GMRES()
202 for (i = 0; i < bn; i++) modul[i] = PetscSqrtReal(wr[i] * wr[i] + wi[i] * wi[i]); in KSPComputeRitz_GMRES()
213 tetar[i] = PetscRealPart(wr[perm[i]]); in KSPComputeRitz_GMRES()
214 tetai[i] = PetscImaginaryPart(wr[perm[i]]); in KSPComputeRitz_GMRES()
217 tetar[i] = PetscRealPart(wr[perm[bn - nb + i]]); in KSPComputeRitz_GMRES()
218 tetai[i] = PetscImaginaryPart(wr[perm[bn - nb + i]]); in KSPComputeRitz_GMRES()
[all …]
|
| /petsc/src/ksp/ksp/impls/gmres/agmres/ |
| H A D | agmresdeflation.c | 98 PetscScalar *wr = agmres->wr; in KSPAGMRESSchurForm() local
127 …", LAPACKhgeqz_("S", "I", "I", &KspSize, &ilo, &KspSize, A, &ldA, B, &ldB, wr, wi, beta, Q, &N, Z,… in KSPAGMRESSchurForm()
130 …gges", LAPACKgges_("V", "V", "N", NULL, &KspSize, A, &ldA, B, &ldB, &sdim, wr, wi, beta, Q, &N, Z,… in KSPAGMRESSchurForm()
137 wr[i] /= beta[i]; in KSPAGMRESSchurForm()
143 PetscCall(KSPAGMRESQuickSort(wr, wi, KspSize, perm)); in KSPAGMRESSchurForm()
158 …", LAPACKtgsen_(&ijob, &wantQ, &wantZ, select, &KspSize, A, &ldA, B, &ldB, wr, wi, beta, Q, &N, Z,… in KSPAGMRESSchurForm()
170 …if (agmres->DeflPrecond) PetscCall(KSPAGMRESLejaOrdering(wr, wi, agmres->Rshift, agmres->Ishift, m… in KSPAGMRESSchurForm()
|
| H A D | agmres.c | 133 PetscCall(KSPAGMRESLejaOrdering(agmres->wr, agmres->wi, agmres->Rshift, agmres->Ishift, max_k)); in KSPComputeShifts_DGMRES()
136 PetscScalar *wr, *wi, *Rshift, *Ishift; in KSPComputeShifts_DGMRES() local
137 PetscCall(PetscMalloc4(2 * max_k, &wr, 2 * max_k, &wi, 2 * max_k, &Rshift, 2 * max_k, &Ishift)); in KSPComputeShifts_DGMRES()
139 wr[i] = agmres->wr[i]; in KSPComputeShifts_DGMRES()
157 wr[max_k + i] = agmres->wr[i]; in KSPComputeShifts_DGMRES()
160 PetscCall(KSPAGMRESLejaOrdering(wr, wi, Rshift, Ishift, 2 * max_k)); in KSPComputeShifts_DGMRES()
166 PetscCall(PetscFree(wr)); in KSPComputeShifts_DGMRES()
501 PetscCall(PetscFree(agmres->wr)); in KSPDestroy_AGMRES()
|
| /petsc/src/ts/impls/glee/ |
| H A D | glee.c | 440 PetscScalar *ws = glee->swork, *wr = glee->rwork; in TSEvaluateStep_GLEE() local
463 for (j = 0; j < r; j++) wr[j] = V[i * r + j]; in TSEvaluateStep_GLEE()
464 PetscCall(VecMAXPY(Y[i], r, wr, glee->X)); in TSEvaluateStep_GLEE()
469 for (j = 0; j < r; j++) wr[j] = F[j]; in TSEvaluateStep_GLEE()
470 PetscCall(VecMAXPY(X, r, wr, Y)); in TSEvaluateStep_GLEE()
478 for (j = 0; j < r; j++) wr[j] = V[i * r + j]; in TSEvaluateStep_GLEE()
479 PetscCall(VecMAXPY(Y[i], r, wr, glee->X)); in TSEvaluateStep_GLEE()
484 for (j = 0; j < r; j++) wr[j] = Fembed[j]; in TSEvaluateStep_GLEE()
485 PetscCall(VecMAXPY(X, r, wr, Y)); in TSEvaluateStep_GLEE()
503 PetscScalar *ws = glee->swork, *wr = glee->rwork; in TSStep_GLEE() local
[all …]
|
| /petsc/src/ksp/ksp/impls/gmres/dgmres/ |
| H A D | dgmres.c | 63 dgmres->wr = NULL; in KSPSetUp_DGMRES()
285 PetscCall(PetscFree(dgmres->wr)); in KSPDestroy_DGMRES()
661 PetscReal *wr, *wi, *modul; /* Real and imaginary part and modulus of the eigenvalues of A */ in KSPDGMRESComputeSchurForm_DGMRES() local
683 if (!dgmres->wr) { in KSPDGMRESComputeSchurForm_DGMRES()
684 PetscCall(PetscMalloc1(n, &dgmres->wr)); in KSPDGMRESComputeSchurForm_DGMRES()
687 wr = dgmres->wr; in KSPDGMRESComputeSchurForm_DGMRES()
722 …PetscCallBLAS("LAPACKhseqr", LAPACKhseqr_("S", "I", &bn, &ilo, &ihi, A, &ldA, wr, wi, Q, &ldQ, wor… in KSPDGMRESComputeSchurForm_DGMRES()
728 for (i = 0; i < n; i++) modul[i] = PetscSqrtReal(wr[i] * wr[i] + wi[i] * wi[i]); in KSPDGMRESComputeSchurForm_DGMRES()
758 …PetscCallBLAS("LAPACKtrsen", LAPACKtrsen_("B", "V", select, &bn, A, &ldA, Q, &ldQ, wr, wi, &NbrEig… in KSPDGMRESComputeSchurForm_DGMRES()
847 …PetscReal *wr, *wi, *beta, *modul; /* Real and imaginary part and modulus of the eigenvalues of… in KSPDGMRESImproveEig_DGMRES() local
[all …]
|
| H A D | dgmresimpl.h | 21 PetscReal *wr, *wi, *modul; /* Real and complex part and modulus of eigenvalues */ \
|
| /petsc/src/ts/tutorials/ |
| H A D | ex11.h | 416 static PetscScalar cl, cr, wl, zl, wr, zr, pst, durl, skpr1, skpr2; in riem1mdt() local
437 wr = *rr * cr; in riem1mdt()
440 *pstar = (wl * *pr + wr * *pl) / (wl + wr); in riem1mdt()
|