Searched refs:Adiag (Results 1 – 3 of 3) sorted by relevance
| /petsc/src/mat/tests/ |
| H A D | ex235.c | 63 Vec Adiag, Aijdiag; in main() local
79 PetscCall(shiftandscale(A, &Adiag)); in main()
86 PetscCall(VecAXPY(Aijdiag, -1.0, Adiag)); in main()
87 PetscCall(VecNorm(Adiag, NORM_2, &dnorm)); in main()
92 PetscCall(VecDestroy(&Adiag)); in main()
|
| /petsc/src/mat/impls/aij/seq/kokkos/ |
| H A D | aijkok.kokkos.cxx | 1047 const auto &Adiag = aijkok->diag_dual.view_device(); in MatShift_SeqAIJKokkos() local
1048 …<>(PetscGetKokkosExecutionSpace(), 0, n), KOKKOS_LAMBDA(const PetscInt i) { Aa(Adiag(i)) += a; })); in MatShift_SeqAIJKokkos()
1076 const auto &Adiag = aijkok->diag_dual.view_device(); in MatDiagonalSet_SeqAIJKokkos() local
1079 if (is == INSERT_VALUES) Aa(Adiag(i)) = dv(i); in MatDiagonalSet_SeqAIJKokkos()
1080 else Aa(Adiag(i)) += dv(i); in MatDiagonalSet_SeqAIJKokkos()
1164 const auto &Adiag = aijkok->diag_dual.view_device(); in MatGetDiagonal_SeqAIJKokkos() local
1169 if (Adiag(i) < Ai(i + 1)) xv(i) = Aa(Adiag(i)); in MatGetDiagonal_SeqAIJKokkos()
1504 auto Adiag = akok->diag_dual.view_device(); in MatInvertVariableBlockDiagonal_SeqAIJKokkos() local
1522 if (Ai(i) <= Adiag(i) && Adiag(i) < Ai(i + 1)) { // if the diagonal exists (common case) in MatInvertVariableBlockDiagonal_SeqAIJKokkos()
1523 …PetscInt first = Adiag(i) - r; // we start to check nonzeros from here along this … in MatInvertVariableBlockDiagonal_SeqAIJKokkos()
|
| /petsc/src/mat/impls/aij/seq/seqhipsparse/ |
| H A D | aijhipsparse.hip.cxx | 1526 const PetscInt *Adiag; in MatILUFactorSymbolic_SeqAIJHIPSPARSE_ILU0() local
1528 PetscCall(MatGetDiagonalMarkers_SeqAIJ(A, &Adiag, NULL)); in MatILUFactorSymbolic_SeqAIJHIPSPARSE_ILU0()
1531 …if (Ai[i] < Adiag[i] && Adiag[i] < Ai[i + 1]) { /* There are nonzeros left to the diagonal of row … in MatILUFactorSymbolic_SeqAIJHIPSPARSE_ILU0()
1533 nzLeft = Adiag[i] - Ai[i]; in MatILUFactorSymbolic_SeqAIJHIPSPARSE_ILU0()
|