Home
last modified time | relevance | path

Searched refs:zin (Results 1 – 4 of 4) sorted by relevance

/petsc/src/vec/vec/impls/seq/
H A Dbvec1.c120 PetscErrorCode VecAXPBYPCZ_Seq(Vec zin, PetscScalar alpha, PetscScalar beta, PetscScalar gamma, Vec… in VecAXPBYPCZ_Seq() argument
122 const PetscInt n = zin->map->n; in VecAXPBYPCZ_Seq()
130 PetscCall(VecGetArray(zin, &zz)); in VecAXPBYPCZ_Seq()
144 PetscCall(VecRestoreArray(zin, &zz)); in VecAXPBYPCZ_Seq()
/petsc/src/vec/vec/impls/seq/seqviennacl/
H A Dvecviennacl.cxx717 PetscErrorCode VecAXPBYPCZ_SeqViennaCL(Vec zin, PetscScalar alpha, PetscScalar beta, PetscScalar ga… in VecAXPBYPCZ_SeqViennaCL() argument
719 PetscInt n = zin->map->n; in VecAXPBYPCZ_SeqViennaCL()
726 PetscCall(VecViennaCLGetArray(zin, &zgpu)); in VecAXPBYPCZ_SeqViennaCL()
785 PetscCall(VecViennaCLRestoreArray(zin, &zgpu)); in VecAXPBYPCZ_SeqViennaCL()
/petsc/src/vec/vec/impls/seq/cupm/
H A Dvecseqcupm_impl.hpp1612 inline PetscErrorCode VecSeq_CUPM<T>::AXPBYPCZAsync(Vec zin, PetscScalar alpha, PetscScalar beta, P… in AXPBYPCZAsync() argument
1616 if (gamma != PetscScalar(1.0)) PetscCall(ScaleAsync(zin, gamma, dctx)); in AXPBYPCZAsync()
1617 PetscCall(AXPYAsync(zin, alpha, xin, dctx)); in AXPBYPCZAsync()
1618 PetscCall(AXPYAsync(zin, beta, yin, dctx)); in AXPBYPCZAsync()
1624 inline PetscErrorCode VecSeq_CUPM<T>::AXPBYPCZ(Vec zin, PetscScalar alpha, PetscScalar beta, PetscS… in AXPBYPCZ() argument
1627 PetscCall(AXPBYPCZAsync(zin, alpha, beta, gamma, xin, yin, nullptr)); in AXPBYPCZ()
/petsc/src/vec/vec/impls/seq/kokkos/
H A Dveckok.kokkos.cxx935 PetscErrorCode VecAXPBYPCZ_SeqKokkos(Vec zin, PetscScalar alpha, PetscScalar beta, PetscScalar gamm… in VecAXPBYPCZ_SeqKokkos() argument
939 Kokkos::RangePolicy<> policy(PetscGetKokkosExecutionSpace(), 0, zin->map->n); in VecAXPBYPCZ_SeqKokkos()
943 PetscCall(VecGetKokkosView(zin, &zv)); in VecAXPBYPCZ_SeqKokkos()
958 PetscCall(VecRestoreKokkosView(zin, &zv)); in VecAXPBYPCZ_SeqKokkos()
960 PetscCall(PetscLogGpuFlops(zin->map->n * 5.0)); in VecAXPBYPCZ_SeqKokkos()