Searched refs:Diag (Results 1 – 2 of 2) sorted by relevance
| /petsc/src/tao/pde_constrained/tutorials/ |
| H A D | elliptic.c | 33 Mat Diag, Ones; member
262 PetscCall(MatDiagonalSet(user->Diag, user->Swork, INSERT_VALUES)); in FormJacobianState()
263 PetscCall(MatPtAP(user->Diag, user->Grad, MAT_REUSE_MATRIX, 1.0, &user->DSG)); in FormJacobianState()
934 PetscCall(MatCreate(PETSC_COMM_WORLD, &user->Diag)); in EllipticInitialize()
935 PetscCall(MatSetSizes(user->Diag, PETSC_DECIDE, PETSC_DECIDE, m, m)); in EllipticInitialize()
936 PetscCall(MatSetFromOptions(user->Diag)); in EllipticInitialize()
937 PetscCall(MatMPIAIJSetPreallocation(user->Diag, 1, NULL, 0, NULL)); in EllipticInitialize()
938 PetscCall(MatSeqAIJSetPreallocation(user->Diag, 1, NULL)); in EllipticInitialize()
981 PetscCall(MatDiagonalSet(user->Diag, user->Swork, INSERT_VALUES)); in EllipticInitialize()
982 PetscCall(MatPtAP(user->Diag, user->Grad, MAT_INITIAL_MATRIX, 1.0, &user->DSG)); in EllipticInitialize()
[all …]
|
| /petsc/src/ksp/pc/impls/bjacobi/bjkokkos/ |
| H A D | bjkokkos.kokkos.cxx | 94 const PetscScalar *Diag = &glb_idiag[start]; in BJSolve_TFQMR() local
197 T[idx] = Diag[idx] * P[idx]; in BJSolve_TFQMR()
221 … parallel_for(Kokkos::TeamVectorRange(team, Nblk), [=](int idx) { T[idx] = Diag[idx] * T[idx]; }); in BJSolve_TFQMR()
300 … parallel_for(Kokkos::TeamVectorRange(team, Nblk), [=](int idx) { T[idx] = Diag[idx] * P[idx]; }); in BJSolve_TFQMR()
311 …parallel_for(Kokkos::TeamVectorRange(team, Nblk), [=](int idx) { XX[idx] = Diag[idx] * XX[idx]; }); in BJSolve_TFQMR()
|