Home
last modified time | relevance | path

Searched refs:yarray (Results 1 – 12 of 12) sorted by relevance

/petsc/src/vec/is/sf/tests/
H A Dex22.c15 PetscScalar *yarray, *zarray; in main() local
39 PetscCall(VecGetArrayAndMemType(y, &yarray, &ymtype)); in main()
42 …PetscCall(PetscSFFetchAndOpWithMemTypeBegin(vscat, MPIU_SCALAR, ymtype, yarray, xmtype, xarray, zm… in main()
43 PetscCall(PetscSFFetchAndOpEnd(vscat, MPIU_SCALAR, yarray, xarray, zarray, MPI_SUM)); in main()
46 PetscCall(VecRestoreArrayAndMemType(y, &yarray)); in main()
/petsc/src/ksp/pc/impls/kaczmarz/
H A Dkaczmarz.c23 PetscScalar *yarray; in PCApply_Kaczmarz() local
31 PetscCall(VecGetArray(y, &yarray)); in PCApply_Kaczmarz()
38 if (cols[j] >= ys && cols[j] < ye) r -= yarray[cols[j] - ys] * vals[j]; in PCApply_Kaczmarz()
43 if (cols[j] >= ys && cols[j] < ye) yarray[cols[j] - ys] += vals[j] * lambda * r / anrm; in PCApply_Kaczmarz()
54 if (cols[j] >= ys && cols[j] < ye) r -= yarray[cols[j] - ys] * vals[j]; in PCApply_Kaczmarz()
59 if (cols[j] >= ys && cols[j] < ye) yarray[cols[j] - ys] += vals[j] * lambda * r / anrm; in PCApply_Kaczmarz()
65 PetscCall(VecRestoreArray(y, &yarray)); in PCApply_Kaczmarz()
/petsc/src/ksp/pc/impls/rowscalingviennacl/
H A Drowscalingviennacl.cxx85 viennacl::vector<PetscScalar> *yarray = NULL; in PCApply_ROWSCALINGVIENNACL() local
95 PetscCall(VecViennaCLGetArrayWrite(y, &yarray)); in PCApply_ROWSCALINGVIENNACL()
100 *yarray = *xarray; in PCApply_ROWSCALINGVIENNACL()
101 ilu->ROWSCALINGVIENNACL->apply(*yarray); in PCApply_ROWSCALINGVIENNACL()
107 PetscCall(VecViennaCLRestoreArrayWrite(y, &yarray)); in PCApply_ROWSCALINGVIENNACL()
/petsc/src/ksp/pc/impls/chowiluviennacl/
H A Dchowiluviennacl.cxx85 viennacl::vector<PetscScalar> *yarray = NULL; in PCApply_CHOWILUVIENNACL() local
95 PetscCall(VecViennaCLGetArrayWrite(y, &yarray)); in PCApply_CHOWILUVIENNACL()
100 *yarray = *xarray; in PCApply_CHOWILUVIENNACL()
101 ilu->CHOWILUVIENNACL->apply(*yarray); in PCApply_CHOWILUVIENNACL()
107 PetscCall(VecViennaCLRestoreArrayWrite(y, &yarray)); in PCApply_CHOWILUVIENNACL()
/petsc/src/ksp/pc/impls/saviennacl/
H A Dsaviennacl.cxx87 viennacl::vector<PetscScalar> *yarray = NULL; in PCApply_SAVIENNACL() local
96 PetscCall(VecViennaCLGetArrayWrite(y, &yarray)); in PCApply_SAVIENNACL()
99 *yarray = *xarray; in PCApply_SAVIENNACL()
100 sac->SAVIENNACL->apply(*yarray); in PCApply_SAVIENNACL()
106 PetscCall(VecViennaCLRestoreArrayWrite(y, &yarray)); in PCApply_SAVIENNACL()
/petsc/src/vec/vec/impls/seq/
H A Dbvec1.c76 PetscScalar *yarray; in VecAXPY_Seq() local
83 PetscCall(VecGetArray(yin, &yarray)); in VecAXPY_Seq()
84 PetscCallBLAS("BLASaxpy", BLASaxpy_(&bn, &alpha, xarray, &one, yarray, &one)); in VecAXPY_Seq()
86 PetscCall(VecRestoreArray(yin, &yarray)); in VecAXPY_Seq()
H A Ddvec2.c518 const PetscScalar *xarray, *yarray, *yfirst, *ynext; in VecMultiDot_Seq_GEMV() local
537 yarray = yfirst; in VecMultiDot_Seq_GEMV()
541 lda = ynext - yarray; in VecMultiDot_Seq_GEMV()
543 } else if (lda * (j - i) != ynext - yarray) { // not in the same stride? if so, stop here in VecMultiDot_Seq_GEMV()
557 …PetscCallBLAS("BLASgemv", BLASgemv_(trans, &n, &m, &one, yarray, &lda2, xarray, &ione, &zero, z + … in VecMultiDot_Seq_GEMV()
705 PetscScalar *yarray; in VecMAXPY_Seq_GEMV() local
718 PetscCall(VecGetArray(yin, &yarray)); in VecMAXPY_Seq_GEMV()
741 …AS("BLASgemv", BLASgemv_("N", &n, &m, &one, xarray, &lda2, alpha + i, &incx, &one, yarray, &incy)); in VecMAXPY_Seq_GEMV()
750 PetscCall(VecRestoreArray(yin, &yarray)); in VecMAXPY_Seq_GEMV()
/petsc/src/mat/impls/baij/seq/
H A Dbaij2.c992 PetscScalar *zarray, *yarray, xv; in MatMultAdd_SeqBAIJ_12_ver1() local
1000 PetscCall(VecGetArrayPair(yy, zz, &yarray, &zarray)); in MatMultAdd_SeqBAIJ_12_ver1()
1004 if (zz != yy) PetscCall(PetscArraycpy(zarray, yarray, 12 * mbs)); in MatMultAdd_SeqBAIJ_12_ver1()
1010 y = yarray; in MatMultAdd_SeqBAIJ_12_ver1()
1019 y = yarray + 12 * ridx[i]; in MatMultAdd_SeqBAIJ_12_ver1()
1074 PetscCall(VecRestoreArrayPair(yy, zz, &yarray, &zarray)); in MatMultAdd_SeqBAIJ_12_ver1()
1197 PetscScalar x1, x2, x3, x4, *zarray, *yarray; in MatMultAdd_SeqBAIJ_12_ver2() local
1205 PetscCall(VecGetArrayPair(yy, zz, &yarray, &zarray)); in MatMultAdd_SeqBAIJ_12_ver2()
1209 if (zz != yy) PetscCall(PetscArraycpy(zarray, yarray, 12 * mbs)); in MatMultAdd_SeqBAIJ_12_ver2()
1215 y = yarray; in MatMultAdd_SeqBAIJ_12_ver2()
[all …]
/petsc/src/mat/impls/scalapack/
H A Dmatscalapack.c250 PetscScalar *yarray; in MatMult_ScaLAPACK() local
254 PetscCall(VecGetArray(y, &yarray)); in MatMult_ScaLAPACK()
255 PetscCall(MatMultXXXYYY_ScaLAPACK(A, PETSC_FALSE, PETSC_FALSE, 0.0, xarray, yarray)); in MatMult_ScaLAPACK()
257 PetscCall(VecRestoreArray(y, &yarray)); in MatMult_ScaLAPACK()
264 PetscScalar *yarray; in MatMultTranspose_ScaLAPACK() local
268 PetscCall(VecGetArray(y, &yarray)); in MatMultTranspose_ScaLAPACK()
269 PetscCall(MatMultXXXYYY_ScaLAPACK(A, PETSC_TRUE, PETSC_FALSE, 0.0, xarray, yarray)); in MatMultTranspose_ScaLAPACK()
271 PetscCall(VecRestoreArray(y, &yarray)); in MatMultTranspose_ScaLAPACK()
278 PetscScalar *yarray; in MatMultHermitianTranspose_ScaLAPACK() local
282 PetscCall(VecGetArrayWrite(y, &yarray)); in MatMultHermitianTranspose_ScaLAPACK()
[all …]
/petsc/src/vec/vec/impls/seq/kokkos/
H A Dveckok.kokkos.cxx534 const PetscScalar *yarray; in VecMultiDot_SeqKokkos_GEMV() local
551 yarray = yfirst.data(); in VecMultiDot_SeqKokkos_GEMV()
555 lda = ynext.data() - yarray; // arbitrary ptrdiff could be very large in VecMultiDot_SeqKokkos_GEMV()
557 …} else if (lda * (j - i) != ynext.data() - yarray) { // not in the same stride? if so, stop search… in VecMultiDot_SeqKokkos_GEMV()
575 const auto &A = Kokkos::View<const PetscScalar **, Kokkos::LayoutLeft>(yarray, lda, m); in VecMultiDot_SeqKokkos_GEMV()
688 PetscScalar *yarray; in VecCopy_SeqKokkos() local
689 PetscCall(VecGetArrayWrite(yin, &yarray)); in VecCopy_SeqKokkos()
690 PetscScalarKokkosViewHost yv(yarray, yin->map->n); in VecCopy_SeqKokkos()
699 PetscCall(VecRestoreArrayWrite(yin, &yarray)); in VecCopy_SeqKokkos()
/petsc/src/ksp/pc/impls/ml/
H A Dml.c152 PetscScalar *yarray; in MatMult_ML() local
159 PetscCall(VecGetArray(y, &yarray)); in MatMult_ML()
162 …erator_Apply", ML_Operator_Apply(shell->mlmat, x_length, (PetscScalar *)xarray, y_length, yarray)); in MatMult_ML()
164 PetscCall(VecRestoreArray(y, &yarray)); in MatMult_ML()
/petsc/src/ksp/pc/impls/bjacobi/
H A Dbjacobi.c1194 PetscScalar *yarray; in PCApply_BJacobi_Multiproc() local
1201 PetscCall(VecGetArray(y, &yarray)); in PCApply_BJacobi_Multiproc()
1203 PetscCall(VecPlaceArray(mpjac->ysub, yarray)); in PCApply_BJacobi_Multiproc()
1216 PetscCall(VecRestoreArray(y, &yarray)); in PCApply_BJacobi_Multiproc()