Lines Matching refs:zz

194 …tic inline PetscErrorCode MatMult_MatMultAdd_SeqMAIJ_Template(Mat A, Vec xx, Vec yy, Vec zz, int N)  in MatMult_MatMultAdd_SeqMAIJ_Template()  argument
211 if (mult_add && yy != zz) PetscCall(VecCopy(yy, zz)); in MatMult_MatMultAdd_SeqMAIJ_Template()
214 PetscCall(VecGetArray(zz, &z)); in MatMult_MatMultAdd_SeqMAIJ_Template()
216 PetscCall(VecGetArrayWrite(zz, &z)); in MatMult_MatMultAdd_SeqMAIJ_Template()
248 PetscCall(VecRestoreArray(zz, &z)); in MatMult_MatMultAdd_SeqMAIJ_Template()
250 PetscCall(VecRestoreArrayWrite(zz, &z)); in MatMult_MatMultAdd_SeqMAIJ_Template()
255 …rorCode MatMultTranspose_MatMultTransposeAdd_SeqMAIJ_Template(Mat A, Vec xx, Vec yy, Vec zz, int N) in MatMultTranspose_MatMultTransposeAdd_SeqMAIJ_Template() argument
272 if (yy != zz) PetscCall(VecCopy(yy, zz)); in MatMultTranspose_MatMultTransposeAdd_SeqMAIJ_Template()
274 PetscCall(VecSet(zz, 0.0)); in MatMultTranspose_MatMultTransposeAdd_SeqMAIJ_Template()
277 PetscCall(VecGetArray(zz, &z)); in MatMultTranspose_MatMultTransposeAdd_SeqMAIJ_Template()
299 PetscCall(VecRestoreArray(zz, &z)); in MatMultTranspose_MatMultTransposeAdd_SeqMAIJ_Template()
316 static PetscErrorCode PetscConcat(MatMultAdd_SeqMAIJ_, N)(Mat A, Vec xx, Vec yy, Vec zz) \
319 PetscCall(MatMult_MatMultAdd_SeqMAIJ_Template(A, xx, yy, zz, N)); \
322 …static PetscErrorCode PetscConcat(MatMultTransposeAdd_SeqMAIJ_, N)(Mat A, Vec xx, Vec yy, Vec zz) \
325 PetscCall(MatMultTranspose_MatMultTransposeAdd_SeqMAIJ_Template(A, xx, yy, zz, N)); \
377 static PetscErrorCode MatMultAdd_SeqMAIJ_N(Mat A, Vec xx, Vec yy, Vec zz) in MatMultAdd_SeqMAIJ_N() argument
387 if (yy != zz) PetscCall(VecCopy(yy, zz)); in MatMultAdd_SeqMAIJ_N()
389 PetscCall(VecGetArray(zz, &y)); in MatMultAdd_SeqMAIJ_N()
406 PetscCall(VecRestoreArray(zz, &y)); in MatMultAdd_SeqMAIJ_N()
440 static PetscErrorCode MatMultTransposeAdd_SeqMAIJ_N(Mat A, Vec xx, Vec yy, Vec zz) in MatMultTransposeAdd_SeqMAIJ_N() argument
450 if (yy != zz) PetscCall(VecCopy(yy, zz)); in MatMultTransposeAdd_SeqMAIJ_N()
452 PetscCall(VecGetArray(zz, &y)); in MatMultTransposeAdd_SeqMAIJ_N()
466 PetscCall(VecRestoreArray(zz, &y)); in MatMultTransposeAdd_SeqMAIJ_N()
495 static PetscErrorCode MatMultAdd_MPIMAIJ_dof(Mat A, Vec xx, Vec yy, Vec zz) in MatMultAdd_MPIMAIJ_dof() argument
502 PetscCall((*b->AIJ->ops->multadd)(b->AIJ, xx, yy, zz)); in MatMultAdd_MPIMAIJ_dof()
504 PetscCall((*b->OAIJ->ops->multadd)(b->OAIJ, b->w, zz, zz)); in MatMultAdd_MPIMAIJ_dof()
508 static PetscErrorCode MatMultTransposeAdd_MPIMAIJ_dof(Mat A, Vec xx, Vec yy, Vec zz) in MatMultTransposeAdd_MPIMAIJ_dof() argument
514 PetscCall((*b->AIJ->ops->multtransposeadd)(b->AIJ, xx, yy, zz)); in MatMultTransposeAdd_MPIMAIJ_dof()
515 PetscCall(VecScatterBegin(b->ctx, b->w, zz, ADD_VALUES, SCATTER_REVERSE)); in MatMultTransposeAdd_MPIMAIJ_dof()
516 PetscCall(VecScatterEnd(b->ctx, b->w, zz, ADD_VALUES, SCATTER_REVERSE)); in MatMultTransposeAdd_MPIMAIJ_dof()