Home
last modified time | relevance | path

Searched refs:y_stride (Results 1 – 3 of 3) sorted by relevance

/petsc/src/ksp/ksp/utils/lmvm/blas_cyclic/cupm/
H A Dblas_cyclic_cupm_impl.hpp32 …r alpha, const PetscScalar x[], PetscScalar beta, PetscScalar y[], cupmBlasInt_t y_stride) noexcept in axpby_dispatch() argument
41 PetscCallCUPMBLAS(cupmBlasXcopy(handle, n, x_, 1, y_, y_stride)); in axpby_dispatch()
43 if (beta != 1.0) PetscCallCUPMBLAS(cupmBlasXscal(handle, n, cbeta, y_, y_stride)); in axpby_dispatch()
44 if (alpha != 0.0) PetscCallCUPMBLAS(cupmBlasXaxpy(handle, n, calpha, x_, 1, y_, y_stride)); in axpby_dispatch()
50 …Scalar alpha, const PetscScalar x[], PetscScalar beta, PetscScalar y[], PetscInt y_stride) noexcept in axpby() argument
62 PetscCall(PetscCUPMBlasIntCast(y_stride, &y_stride_)); in axpby()
72 …PetscCall(axpby_dispatch(handle, diff, alpha, &x[i_oldest], beta, &y[i_oldest * y_stride], y_strid… in axpby()
77 …cCall(axpby_dispatch(handle, diff, alpha, &x[i_oldest], beta, &y[i_oldest * y_stride], y_stride_)); in axpby()
H A Dblas_cyclic_cupm.cxx30 …xt, PetscScalar alpha, const PetscScalar x[], PetscScalar beta, PetscScalar y[], PetscInt y_stride) in AXPBYCyclic_CUPM_Private() argument
41 …Petsc::device::cupm::DeviceType::CUDA>::axpby(dctx, m, oldest, next, alpha, x, beta, y, y_stride)); in AXPBYCyclic_CUPM_Private()
46 …:Petsc::device::cupm::DeviceType::HIP>::axpby(dctx, m, oldest, next, alpha, x, beta, y, y_stride)); in AXPBYCyclic_CUPM_Private()
/petsc/src/ksp/ksp/utils/lmvm/blas_cyclic/
H A Dblas_cyclic.c16 … m, PetscScalar alpha, const PetscScalar x[], PetscScalar beta, PetscScalar y[], PetscInt y_stride) in AXPBY_Private() argument
18 for (PetscInt i = 0; i < m; i++) y[i * y_stride] = alpha * x[i] + beta * y[i * y_stride]; in AXPBY_Private()
21 …xt, PetscScalar alpha, const PetscScalar x[], PetscScalar beta, PetscScalar y[], PetscInt y_stride) in AXPBYCylic_Private() argument
28 AXPBY_Private(m, alpha, x, beta, y, y_stride); in AXPBYCylic_Private()
30 AXPBY_Private(i_next - i_oldest, alpha, &x[i_oldest], beta, &y[i_oldest * y_stride], y_stride); in AXPBYCylic_Private()
32 AXPBY_Private(i_next, alpha, x, beta, y, y_stride); in AXPBYCylic_Private()
33 AXPBY_Private(m - i_oldest, alpha, &x[i_oldest], beta, &y[i_oldest * y_stride], y_stride); in AXPBYCylic_Private()