Lines Matching refs:yy
194 …tic inline PetscErrorCode MatMult_MatMultAdd_SeqMAIJ_Template(Mat A, Vec xx, Vec yy, Vec zz, int N) in MatMult_MatMultAdd_SeqMAIJ_Template() argument
196 const PetscBool mult_add = yy == NULL ? PETSC_FALSE : PETSC_TRUE; in MatMult_MatMultAdd_SeqMAIJ_Template()
211 if (mult_add && yy != zz) PetscCall(VecCopy(yy, zz)); 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
257 const PetscBool mult_add = yy == NULL ? PETSC_FALSE : PETSC_TRUE; in MatMultTranspose_MatMultTransposeAdd_SeqMAIJ_Template()
272 if (yy != zz) PetscCall(VecCopy(yy, zz)); in MatMultTranspose_MatMultTransposeAdd_SeqMAIJ_Template()
304 static PetscErrorCode PetscConcat(MatMult_SeqMAIJ_, N)(Mat A, Vec xx, Vec yy) \
307 PetscCall(MatMult_MatMultAdd_SeqMAIJ_Template(A, xx, NULL, yy, N)); \
310 static PetscErrorCode PetscConcat(MatMultTranspose_SeqMAIJ_, N)(Mat A, Vec xx, Vec yy) \
313 PetscCall(MatMultTranspose_MatMultTransposeAdd_SeqMAIJ_Template(A, xx, NULL, yy, N)); \
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)); \
344 static PetscErrorCode MatMult_SeqMAIJ_N(Mat A, Vec xx, Vec yy) in MatMult_SeqMAIJ_N() argument
355 PetscCall(VecSet(yy, 0.0)); in MatMult_SeqMAIJ_N()
356 PetscCall(VecGetArray(yy, &y)); in MatMult_SeqMAIJ_N()
373 PetscCall(VecRestoreArray(yy, &y)); in MatMult_SeqMAIJ_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()
410 static PetscErrorCode MatMultTranspose_SeqMAIJ_N(Mat A, Vec xx, Vec yy) in MatMultTranspose_SeqMAIJ_N() argument
421 PetscCall(VecSet(yy, 0.0)); in MatMultTranspose_SeqMAIJ_N()
422 PetscCall(VecGetArray(yy, &y)); in MatMultTranspose_SeqMAIJ_N()
436 PetscCall(VecRestoreArray(yy, &y)); in MatMultTranspose_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()
470 static PetscErrorCode MatMult_MPIMAIJ_dof(Mat A, Vec xx, Vec yy) in MatMult_MPIMAIJ_dof() argument
477 PetscCall((*b->AIJ->ops->mult)(b->AIJ, xx, yy)); in MatMult_MPIMAIJ_dof()
479 PetscCall((*b->OAIJ->ops->multadd)(b->OAIJ, b->w, yy, yy)); in MatMult_MPIMAIJ_dof()
483 static PetscErrorCode MatMultTranspose_MPIMAIJ_dof(Mat A, Vec xx, Vec yy) in MatMultTranspose_MPIMAIJ_dof() argument
489 PetscCall((*b->AIJ->ops->multtranspose)(b->AIJ, xx, yy)); in MatMultTranspose_MPIMAIJ_dof()
490 PetscCall(VecScatterBegin(b->ctx, b->w, yy, ADD_VALUES, SCATTER_REVERSE)); in MatMultTranspose_MPIMAIJ_dof()
491 PetscCall(VecScatterEnd(b->ctx, b->w, yy, ADD_VALUES, SCATTER_REVERSE)); in MatMultTranspose_MPIMAIJ_dof()
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()
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()