Lines Matching refs:ax

476   const PetscScalar *ax;  in MatMultColumnRange_MPIDense()  local
482 PetscCall(VecGetArrayReadAndMemType(xx, &ax, &axmtype)); in MatMultColumnRange_MPIDense()
484 …PetscCall(PetscSFBcastWithMemTypeBegin(mdn->Mvctx, MPIU_SCALAR, axmtype, ax, aymtype, ay, MPI_REPL… in MatMultColumnRange_MPIDense()
485 PetscCall(PetscSFBcastEnd(mdn->Mvctx, MPIU_SCALAR, ax, ay, MPI_REPLACE)); in MatMultColumnRange_MPIDense()
487 PetscCall(VecRestoreArrayReadAndMemType(xx, &ax)); in MatMultColumnRange_MPIDense()
495 const PetscScalar *ax; in MatMult_MPIDense() local
501 PetscCall(VecGetArrayReadAndMemType(xx, &ax, &axmtype)); in MatMult_MPIDense()
503 …PetscCall(PetscSFBcastWithMemTypeBegin(mdn->Mvctx, MPIU_SCALAR, axmtype, ax, aymtype, ay, MPI_REPL… in MatMult_MPIDense()
504 PetscCall(PetscSFBcastEnd(mdn->Mvctx, MPIU_SCALAR, ax, ay, MPI_REPLACE)); in MatMult_MPIDense()
506 PetscCall(VecRestoreArrayReadAndMemType(xx, &ax)); in MatMult_MPIDense()
514 const PetscScalar *ax; in MatMultAddColumnRange_MPIDense() local
520 PetscCall(VecGetArrayReadAndMemType(xx, &ax, &axmtype)); in MatMultAddColumnRange_MPIDense()
522 …PetscCall(PetscSFBcastWithMemTypeBegin(mdn->Mvctx, MPIU_SCALAR, axmtype, ax, aymtype, ay, MPI_REPL… in MatMultAddColumnRange_MPIDense()
523 PetscCall(PetscSFBcastEnd(mdn->Mvctx, MPIU_SCALAR, ax, ay, MPI_REPLACE)); in MatMultAddColumnRange_MPIDense()
525 PetscCall(VecRestoreArrayReadAndMemType(xx, &ax)); in MatMultAddColumnRange_MPIDense()
533 const PetscScalar *ax; in MatMultAdd_MPIDense() local
539 PetscCall(VecGetArrayReadAndMemType(xx, &ax, &axmtype)); in MatMultAdd_MPIDense()
541 …PetscCall(PetscSFBcastWithMemTypeBegin(mdn->Mvctx, MPIU_SCALAR, axmtype, ax, aymtype, ay, MPI_REPL… in MatMultAdd_MPIDense()
542 PetscCall(PetscSFBcastEnd(mdn->Mvctx, MPIU_SCALAR, ax, ay, MPI_REPLACE)); in MatMultAdd_MPIDense()
544 PetscCall(VecRestoreArrayReadAndMemType(xx, &ax)); in MatMultAdd_MPIDense()
552 const PetscScalar *ax; in MatMultHermitianTransposeColumnRange_MPIDense() local
574 PetscCall(VecGetArrayReadAndMemType(a->lvec, &ax, &axmtype)); in MatMultHermitianTransposeColumnRange_MPIDense()
575 …PetscCall(PetscSFReduceWithMemTypeBegin(a->Mvctx, MPIU_SCALAR, axmtype, ax, aymtype, ay, MPIU_SUM)… in MatMultHermitianTransposeColumnRange_MPIDense()
576 PetscCall(PetscSFReduceEnd(a->Mvctx, MPIU_SCALAR, ax, ay, MPIU_SUM)); in MatMultHermitianTransposeColumnRange_MPIDense()
577 PetscCall(VecRestoreArrayReadAndMemType(a->lvec, &ax)); in MatMultHermitianTransposeColumnRange_MPIDense()
585 const PetscScalar *ax; in MatMultTransposeKernel_MPIDense() local
594 PetscCall(VecGetArrayReadAndMemType(a->lvec, &ax, &axmtype)); in MatMultTransposeKernel_MPIDense()
596 …PetscCall(PetscSFReduceWithMemTypeBegin(a->Mvctx, MPIU_SCALAR, axmtype, ax, aymtype, ay, MPIU_SUM)… in MatMultTransposeKernel_MPIDense()
597 PetscCall(PetscSFReduceEnd(a->Mvctx, MPIU_SCALAR, ax, ay, MPIU_SUM)); in MatMultTransposeKernel_MPIDense()
598 PetscCall(VecRestoreArrayReadAndMemType(a->lvec, &ax)); in MatMultTransposeKernel_MPIDense()
606 const PetscScalar *ax; in MatMultHermitianTransposeAddColumnRange_MPIDense() local
615 PetscCall(VecGetArrayReadAndMemType(a->lvec, &ax, &axmtype)); in MatMultHermitianTransposeAddColumnRange_MPIDense()
617 …PetscCall(PetscSFReduceWithMemTypeBegin(a->Mvctx, MPIU_SCALAR, axmtype, ax, aymtype, ay, MPIU_SUM)… in MatMultHermitianTransposeAddColumnRange_MPIDense()
618 PetscCall(PetscSFReduceEnd(a->Mvctx, MPIU_SCALAR, ax, ay, MPIU_SUM)); in MatMultHermitianTransposeAddColumnRange_MPIDense()
619 PetscCall(VecRestoreArrayReadAndMemType(a->lvec, &ax)); in MatMultHermitianTransposeAddColumnRange_MPIDense()
627 const PetscScalar *ax; in MatMultTransposeAddKernel_MPIDense() local
636 PetscCall(VecGetArrayReadAndMemType(a->lvec, &ax, &axmtype)); in MatMultTransposeAddKernel_MPIDense()
638 …PetscCall(PetscSFReduceWithMemTypeBegin(a->Mvctx, MPIU_SCALAR, axmtype, ax, aymtype, ay, MPIU_SUM)… in MatMultTransposeAddKernel_MPIDense()
639 PetscCall(PetscSFReduceEnd(a->Mvctx, MPIU_SCALAR, ax, ay, MPIU_SUM)); in MatMultTransposeAddKernel_MPIDense()
640 PetscCall(VecRestoreArrayReadAndMemType(a->lvec, &ax)); in MatMultTransposeAddKernel_MPIDense()