Searched refs:yidx (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/vec/vec/impls/seq/cupm/ |
| H A D | vecseqcupm_impl.hpp | 924 …PetscScalar *xptr, const PetscScalar *aptr, const Vec *yin, PetscInt size, PetscInt &yidx) noexcept in MAXPY_kernel_dispatch_() argument
927 …PetscCall(MAXPY_kernel_dispatch_(dctx, stream, xptr, aptr + yidx, yin + yidx, size, util::make_ind… in MAXPY_kernel_dispatch_()
928 yidx += N; in MAXPY_kernel_dispatch_()
951 PetscInt yidx = 0; in MAXPYAsync() local
960 switch (nv - yidx) { in MAXPYAsync()
962 PetscCall(MAXPY_kernel_dispatch_<7>(dctx, stream, xptr.data(), d_alpha, yin, n, yidx)); in MAXPYAsync()
965 PetscCall(MAXPY_kernel_dispatch_<6>(dctx, stream, xptr.data(), d_alpha, yin, n, yidx)); in MAXPYAsync()
968 PetscCall(MAXPY_kernel_dispatch_<5>(dctx, stream, xptr.data(), d_alpha, yin, n, yidx)); in MAXPYAsync()
971 PetscCall(MAXPY_kernel_dispatch_<4>(dctx, stream, xptr.data(), d_alpha, yin, n, yidx)); in MAXPYAsync()
974 PetscCall(MAXPY_kernel_dispatch_<3>(dctx, stream, xptr.data(), d_alpha, yin, n, yidx)); in MAXPYAsync()
[all …]
|