Home
last modified time | relevance | path

Searched refs:cupmStream_t (Results 1 – 15 of 15) sorted by relevance

/petsc/include/petsc/private/
H A Dcupmobject.hpp199 …leDispatch_(PetscDeviceContext, cupmBlasHandle_t *, cupmSolverHandle_t *, cupmStream_t *) noexcept;
200 …Dispatch_(PetscDeviceContext *, cupmBlasHandle_t *, cupmSolverHandle_t *, cupmStream_t *) noexcept;
209 …text *, cupmBlasHandle_t * = nullptr, cupmSolverHandle_t * = nullptr, cupmStream_t * = nullptr) no…
212 …static PetscErrorCode GetHandles_(PetscDeviceContext *, cupmBlasHandle_t *, cupmStream_t *) noexce…
213 …static PetscErrorCode GetHandles_(PetscDeviceContext *, cupmSolverHandle_t *, cupmStream_t *) noex…
217 static PetscErrorCode GetHandles_(PetscDeviceContext *, cupmStream_t *) noexcept;
222 static PetscErrorCode GetHandles_(cupmStream_t *) noexcept;
224 …tscDeviceContext, cupmBlasHandle_t *, cupmSolverHandle_t * = nullptr, cupmStream_t * = nullptr) no…
225 …static PetscErrorCode GetHandlesFrom_(PetscDeviceContext, cupmSolverHandle_t *, cupmStream_t * = n…
226 static PetscErrorCode GetHandlesFrom_(PetscDeviceContext, cupmStream_t *) noexcept;
[all …]
H A Dcupminterface.hpp146 using cupmStream_t = cudaStream_t; typedef
292 using cupmStream_t = hipStream_t; typedef
446 …using cupmStream_t = typename ::Petsc::device::cupm::impl::InterfaceImpl<T>::cupmStream…
630 …static PetscErrorCode PetscCUPMMallocAsync(M **ptr, std::size_t n, cupmStream_t stream = nullptr) … in PetscCUPMMallocAsync()
674 … util::type_identity_t<D> *src, std::size_t n, cupmMemcpyKind_t kind, cupmStream_t stream = nullpt… in PetscCUPMMemcpyAsync()
717 …_pitch, std::size_t width, std::size_t height, cupmMemcpyKind_t kind, cupmStream_t stream = nullpt… in PetscCUPMMemcpy2DAsync()
747 …static PetscErrorCode PetscCUPMMemsetAsync(M *ptr, int value, std::size_t n, cupmStream_t stream =… in PetscCUPMMemsetAsync()
774 … std::size_t pitch, int value, std::size_t width, std::size_t height, cupmStream_t stream = nullpt… in PetscCUPMMemset2DAsync()
794 PETSC_NODISCARD static cupmError_t cupmFreeAsync(M &ptr, cupmStream_t stream = nullptr) noexcept in cupmFreeAsync()
808 …PETSC_NODISCARD static cupmError_t cupmFreeAsync(std::nullptr_t ptr, cupmStream_t stream = nullptr… in cupmFreeAsync()
[all …]
H A Dveccupmimpl.h305 cupmStream_t stream; in ResetAllocatedDevicePtr_()
400 cupmStream_t stream; in DeviceAllocateCheck_()
419 cupmStream_t stream; in CopyToDevice_()
436 cupmStream_t stream; in CopyToHost_()
601 cupmStream_t stream; in GetArray()
1074 cupmStream_t stream; in ResetPreallocationCOO_CUPMBase()
1113 cupmStream_t stream; in SetPreallocationCOO_CUPMBase()
H A Dmatdensecupmimpl.h411 cupmStream_t stream; in DiagonalUnaryTransform()
472 cupmStream_t stream; in GetDiagonal()
/petsc/src/sys/objects/device/impls/cupm/
H A Dcupmstream.hpp31 using stream_type = cupmStream_t;
61 stream_ = cupmStream_t{}; in destroy()
H A Dcupmevent.hpp118 PetscErrorCode record(cupmStream_t) noexcept;
161 inline PetscErrorCode CUPMEvent<T>::record(cupmStream_t stream) noexcept in record()
H A Dcupmcontext.hpp37 using stream_tag = HandleTag<cupmStream_t, 0>;
68 …PETSC_NODISCARD const cupmStream_t &get(stream_tag) const noexcept { return this->stream.get_strea… in get()
/petsc/src/mat/impls/dense/seq/cupm/
H A Dmatseqdensecupm.hpp55 static PetscErrorCode CheckCUPMSolverInfo_(const cupmBlasInt_t *, cupmStream_t) noexcept;
208 cupmStream_t stream; in SetPreallocation_()
246 cupmStream_t stream; in HostToDevice_()
283 cupmStream_t stream; in DeviceToHost_()
308 …nse_Seq_CUPM<T>::CheckCUPMSolverInfo_(const cupmBlasInt_t *fact_info, cupmStream_t stream) noexcept in CheckCUPMSolverInfo_()
334 …static PetscErrorCode ResizeFactLwork(Mat_SeqDenseCUPM *mcu, cupmStream_t stream, F &&cupmSolverCo… in ResizeFactLwork()
348 static PetscErrorCode FactorPrepare(Mat A, cupmStream_t stream) noexcept in FactorPrepare()
377 cupmStream_t stream; in Factor()
413 …nt_t m, cupmBlasInt_t nrhs, cupmBlasInt_t k, PetscDeviceContext dctx, cupmStream_t stream) noexcept in Solve()
463 cupmStream_t stream; in Factor()
[all …]
/petsc/src/tao/unconstrained/tutorials/
H A Drosenbrock4.h307 typedef cudaStream_t cupmStream_t; typedef
319 typedef hipStream_t cupmStream_t; typedef
360 static PetscErrorCode RosenbrockObjective_Device(cupmStream_t stream, Rosenbrock r, const PetscScal… in RosenbrockObjective_Device()
370 static PetscErrorCode RosenbrockGradient_Device(cupmStream_t stream, Rosenbrock r, const PetscScala… in RosenbrockGradient_Device()
380 static PetscErrorCode RosenbrockObjectiveGradient_Device(cupmStream_t stream, Rosenbrock r, const P… in RosenbrockObjectiveGradient_Device()
390 static PetscErrorCode RosenbrockHessian_Device(cupmStream_t stream, Rosenbrock r, const PetscScalar… in RosenbrockHessian_Device()
486 cupmStream_t *stream; in FormObjective()
524 cupmStream_t *stream; in FormGradient()
583 cupmStream_t *stream; in FormObjectiveGradient()
640 cupmStream_t *stream; in FormHessian()
/petsc/src/vec/vec/impls/seq/cupm/
H A Dvecseqcupm_impl.hpp189 cupmStream_t stream; in PointwiseBinary_()
237 cupmStream_t stream; in PointwiseUnary_()
370 cupmStream_t stream; in GetLocalVector()
430 cupmStream_t stream; in RestoreLocalVector()
459 cupmStream_t stream; in AYPXAsync()
820 cupmStream_t stream; in WAXPYAsync()
907 inline PetscErrorCode VecSeq_CUPM<T>::MAXPY_kernel_dispatch_(PetscDeviceContext dctx, cupmStream_t … in MAXPY_kernel_dispatch_()
924 inline PetscErrorCode VecSeq_CUPM<T>::MAXPY_kernel_dispatch_(PetscDeviceContext dctx, cupmStream_t … in MAXPY_kernel_dispatch_()
937 cupmStream_t stream; in MAXPYAsync()
1152 inline PetscErrorCode VecSeq_CUPM<T>::MDot_kernel_dispatch_(PetscDeviceContext dctx, cupmStream_t s… in MDot_kernel_dispatch_()
[all …]
H A Dvecseqcupm.hpp56 …static PetscErrorCode MDot_kernel_dispatch_(PetscDeviceContext, cupmStream_t, const PetscScalar *,…
58 …static PetscErrorCode MDot_kernel_dispatch_(PetscDeviceContext, cupmStream_t, const PetscScalar *,…
60 …static PetscErrorCode MAXPY_kernel_dispatch_(PetscDeviceContext, cupmStream_t, PetscScalar *, cons…
62 …static PetscErrorCode MAXPY_kernel_dispatch_(PetscDeviceContext, cupmStream_t, PetscScalar *, cons…
/petsc/src/ksp/ksp/utils/lmvm/blas_cyclic/cupm/
H A Dblas_cyclic_cupm_impl.hpp135 cupmStream_t stream; in dsv()
185 cupmStream_t stream; in trsv()
228 cupmStream_t stream; in hemv()
271 cupmStream_t stream; in gemv()
/petsc/src/vec/is/sf/impls/basic/
H A Dsfpack.h6 typedef cudaStream_t cupmStream_t; typedef
12 typedef hipStream_t cupmStream_t; typedef
168 …cupmStream_t stream; /* stream on which input/output root/leafdata is computed o…
202 cupmStream_t remoteCommStream; /* Streams for remote (i.e., inter-rank) communication */
/petsc/src/vec/vec/impls/mpi/cupm/
H A Dvecmpicupm_impl.hpp322 cupmStream_t stream; in SetValuesCOO()
/petsc/src/vec/is/sf/impls/basic/cupm/
H A Dsfcupm_impl.hpp784 cupmStream_t *stream; in LinkSetUp()