Searched refs:Zr (Results 1 – 2 of 2) sorted by relevance
| /petsc/src/ksp/ksp/impls/bicg/ |
| H A D | bicg.c | 16 Vec X, B, Zl, Zr, Rl, Rr, Pl, Pr; in KSPSolve_BiCG() local
29 Zr = ksp->work[4]; in KSPSolve_BiCG()
41 PetscCall(KSP_PCApply(ksp, Rr, Zr)); /* z <- Br */ in KSPSolve_BiCG()
44 PetscCall(VecNorm(Zr, NORM_2, &dp)); /* dp <- z'*z */ in KSPSolve_BiCG()
61 PetscCall(VecDot(Zr, Rl, &beta)); /* beta <- r'z */ in KSPSolve_BiCG()
68 PetscCall(VecCopy(Zr, Pr)); /* p <- z */ in KSPSolve_BiCG()
72 PetscCall(VecAYPX(Pr, b, Zr)); /* p <- z + b* p */ in KSPSolve_BiCG()
77 PetscCall(KSP_MatMult(ksp, Amat, Pr, Zr)); /* z <- Kp */ in KSPSolve_BiCG()
79 PetscCall(VecDot(Zr, Pl, &dpi)); /* dpi <- z'p */ in KSPSolve_BiCG()
84 PetscCall(VecAXPY(Rr, ma, Zr)); in KSPSolve_BiCG()
[all …]
|
| /petsc/src/ksp/pc/impls/bjacobi/bjkokkos/ |
| H A D | bjkokkos.kokkos.cxx | 373 PetscScalar *Zr = ptr; in BJSolve_BICG() local
391 Zr[idx] = Di[idx] * Rr[idx]; in BJSolve_BICG()
416 …TeamVectorRange(team, Nblk), [=](const int idx, PetscScalar &dot) { dot += Zr[idx] * PetscConj(Rl[… in BJSolve_BICG()
430 Pr[idx] = Zr[idx]; in BJSolve_BICG()
438 Pr[idx] = t1 * Pr[idx] + Zr[idx]; in BJSolve_BICG()
445 static_cast<void>(MatMult(team, glb_Aai, glb_Aaj, glb_Aaa, r, ic, start, end, Pr, Zr)); in BJSolve_BICG()
448 …amVectorRange(team, Nblk), [=](const int idx, PetscScalar &lsum) { lsum += Zr[idx] * PetscConj(Pl[… in BJSolve_BICG()
461 Rr[idx] = Rr[idx] + t1 * Zr[idx]; in BJSolve_BICG()
502 Zr[idx] = Di[idx] * Rr[idx]; in BJSolve_BICG()
|