Home
last modified time | relevance | path

Searched refs:cupmScalarPtrCast (Results 1 – 6 of 6) sorted by relevance

/petsc/src/ksp/ksp/utils/lmvm/blas_cyclic/cupm/
H A Dblas_cyclic_cupm_impl.hpp34 auto x_ = cupmScalarPtrCast(x); in axpby_dispatch()
35 auto y_ = cupmScalarPtrCast(y); in axpby_dispatch()
36 const auto calpha = cupmScalarPtrCast(&alpha); in axpby_dispatch()
37 const auto cbeta = cupmScalarPtrCast(&beta); in axpby_dispatch()
93 const auto A_ = cupmScalarPtrCast(A); in dmv()
94 const auto x_ = cupmScalarPtrCast(x); in dmv()
95 const auto y_ = cupmScalarPtrCast(y); in dmv()
96 const auto calpha = cupmScalarPtrCast(&alpha); in dmv()
97 const auto cbeta = cupmScalarPtrCast(&beta); in dmv()
136 const auto A_ = cupmScalarPtrCast(A); in dsv()
[all …]
/petsc/src/ksp/ksp/utils/lmvm/dense/cd_cupm/
H A Dcd_cupm.cxx30 auto A_ = cupmScalarPtrCast(A); in SolveInPlace()
31 auto x_ = cupmScalarPtrCast(x); in SolveInPlace()
56 auto A_ = cupmScalarPtrCast(A); in SolveInPlaceCyclic()
57 auto x_ = cupmScalarPtrCast(x); in SolveInPlaceCyclic()
/petsc/src/vec/vec/impls/seq/cupm/
H A Dvecseqcupm_impl.hpp467 const auto calpha = cupmScalarPtrCast(&alpha); in AYPXAsync()
516 …PetscCallCUPMBLAS(cupmBlasXaxpy(cupmBlasHandle, n, cupmScalarPtrCast(&alpha), DeviceArrayRead(dctx… in AXPYAsync()
835 …PetscCallCUPMBLAS(cupmBlasXaxpy(cupmBlasHandle, n, cupmScalarPtrCast(&alpha), DeviceArrayRead(dctx… in WAXPYAsync()
1023 …lasHandle, n, DeviceArrayRead(dctx, yin), 1, DeviceArrayRead(dctx, xin), 1, cupmScalarPtrCast(z))); in Dot()
1289 …sXdot(handle, n, DeviceArrayRead(sub, yin[i]), 1, xptr.cupmdata(), 1, cupmScalarPtrCast(d_z + i))); in MDot_()
1371 …PetscCallCUPMBLAS(cupmBlasXscal(cupmBlasHandle, n, cupmScalarPtrCast(&alpha), DeviceArrayReadWrite… in ScaleAsync()
1408 …lasHandle, n, DeviceArrayRead(dctx, xin), 1, DeviceArrayRead(dctx, yin), 1, cupmScalarPtrCast(z))); in TDot()
1568 const auto aptr = cupmScalarPtrCast(&alpha); in AXPBYAsync()
1588 … PetscCallCUPMBLAS(cupmBlasXscal(cupmBlasHandle, n, cupmScalarPtrCast(&beta), yptr.cupmdata(), 1)); in AXPBYAsync()
/petsc/include/petsc/private/
H A Dcupminterface.hpp553 …FINED_REFERENCE_BUG_WORKAROUND static constexpr const cupmScalar_t *cupmScalarPtrCast(const PetscS… in cupmScalarPtrCast() function
555 …R_UNDEFINED_REFERENCE_BUG_WORKAROUND static constexpr cupmScalar_t *cupmScalarPtrCast(PetscScalar … in cupmScalarPtrCast() function
931 using ::Petsc::device::cupm::impl::Interface<T>::cupmScalarPtrCast; \
H A Dcupmobject.hpp173 return cupmScalarPtrCast(data()); in cupmdata()
/petsc/src/mat/impls/dense/seq/cupm/
H A Dmatseqdensecupm.hpp689 …PetscCall(Solver{}.template Solve<transpose>(A, cupmScalarPtrCast(y_array), m, m, 1, k, dctx, stre… in MatSolve_Factored_Dispatch_()
781 …PetscCall(Solver{}.template Solve<transpose>(A, cupmScalarPtrCast(y), ldy, m, nrhs, k, dctx, strea… in MatMatSolve_Factored_Dispatch_()