Searched refs:yptr (Results 1 – 2 of 2) sorted by relevance
| /petsc/src/vec/vec/impls/seq/ |
| H A D | dvec2.c | 667 const PetscScalar *yptr[4]; in VecMAXPY_Seq() local
670 #pragma disjoint(*xx, **yptr, *aptr) in VecMAXPY_Seq()
676 for (PetscInt i = 0; i < j_rem; ++i) PetscCall(VecGetArrayRead(y[i], yptr + i)); in VecMAXPY_Seq()
679 PetscKernelAXPY3(xx, alpha[0], alpha[1], alpha[2], yptr[0], yptr[1], yptr[2], n); in VecMAXPY_Seq()
682 PetscKernelAXPY2(xx, alpha[0], alpha[1], yptr[0], yptr[1], n); in VecMAXPY_Seq()
685 PetscKernelAXPY(xx, alpha[0], yptr[0], n); in VecMAXPY_Seq()
689 for (PetscInt i = 0; i < j_rem; ++i) PetscCall(VecRestoreArrayRead(y[i], yptr + i)); in VecMAXPY_Seq()
693 for (PetscInt i = 0; i < inc; ++i) PetscCall(VecGetArrayRead(y[i], yptr + i)); in VecMAXPY_Seq()
694 …PetscKernelAXPY4(xx, alpha[0], alpha[1], alpha[2], alpha[3], yptr[0], yptr[1], yptr[2], yptr[3], n… in VecMAXPY_Seq()
695 for (PetscInt i = 0; i < inc; ++i) PetscCall(VecRestoreArrayRead(y[i], yptr + i)); in VecMAXPY_Seq()
|
| /petsc/src/vec/vec/impls/seq/cupm/ |
| H A D | vecseqcupm_impl.hpp | 472 const auto yptr = DeviceArrayReadWrite(dctx, yin); in AYPXAsync() local
477 …PetscCallCUPMBLAS(cupmBlasXaxpy(cupmBlasHandle, n, calpha, xptr.cupmdata(), 1, yptr.cupmdata(), 1)… in AYPXAsync()
481 PetscCallCUPMBLAS(cupmBlasXscal(cupmBlasHandle, n, calpha, yptr.cupmdata(), 1)); in AYPXAsync()
482 … PetscCallCUPMBLAS(cupmBlasXaxpy(cupmBlasHandle, n, &one, xptr.cupmdata(), 1, yptr.cupmdata(), 1)); in AYPXAsync()
857 …scInt size, PetscScalar *PETSC_RESTRICT xptr, const PetscScalar *PETSC_RESTRICT aptr, Args... yptr) in MAXPY_kernel() argument
861 const PetscScalar *yptr_p[] = {yptr...}; in MAXPY_kernel()
1467 const auto yptr = DeviceArrayWrite(dctx, yout); in CopyAsync() local
1471 PetscCall(PetscCUPMMemcpyAsync(yptr.data(), xptr, n, mode, stream)); in CopyAsync()
1477 PetscScalar *yptr; in CopyAsync() local
1479 PetscCall(VecGetArrayWrite(yout, &yptr)); in CopyAsync()
[all …]
|