Home
last modified time | relevance | path

Searched refs:cupmBlasHandle_t (Results 1 – 7 of 7) sorted by relevance

/petsc/include/petsc/private/
H A Dcupmobject.hpp199 …static PetscErrorCode GetFromHandleDispatch_(PetscDeviceContext, cupmBlasHandle_t *, cupmSolverHan…
200 …static PetscErrorCode GetHandleDispatch_(PetscDeviceContext *, cupmBlasHandle_t *, cupmSolverHandl…
209 …static PetscErrorCode GetHandles_(PetscDeviceContext *, cupmBlasHandle_t * = nullptr, cupmSolverHa…
212 …static PetscErrorCode GetHandles_(PetscDeviceContext *, cupmBlasHandle_t *, cupmStream_t *) noexce…
220 static PetscErrorCode GetHandles_(cupmBlasHandle_t *) noexcept;
224 …static PetscErrorCode GetHandlesFrom_(PetscDeviceContext, cupmBlasHandle_t *, cupmSolverHandle_t *…
247 …Code CUPMObject<T>::GetFromHandleDispatch_(PetscDeviceContext dctx, cupmBlasHandle_t *blas_handle,… in GetFromHandleDispatch_()
281 inline PetscErrorCode CUPMObject<T>::GetHandleDispatch_(PetscDeviceContext *dctx, cupmBlasHandle_t … in GetHandleDispatch_()
295 inline PetscErrorCode CUPMObject<T>::GetHandles_(PetscDeviceContext *dctx, cupmBlasHandle_t *blas_h… in GetHandles_()
301 inline PetscErrorCode CUPMObject<T>::GetHandles_(PetscDeviceContext *dctx, cupmBlasHandle_t *blas_h… in GetHandles_()
[all …]
H A Dcupmblasinterface.hpp236 using cupmBlasHandle_t = cupmBlasHandleWrapper<cublasHandle_t, 0>; typedef
308 using cupmBlasHandle_t = cupmBlasHandleWrapper<hipblasHandle_t, 0>; typedef
376 …using cupmBlasHandle_t = typename ::Petsc::device::cupm::impl::BlasInterfaceImpl<T>::cupmBlas…
431 …static PetscErrorCode PetscCUPMBlasSetPointerModeFromPointer(cupmBlasHandle_t handle, const void *… in PetscCUPMBlasSetPointerModeFromPointer()
459 …CUPMBlasPointerModeGuard(const cupmBlasHandle_t &handle, cupmBlasPointerMode_t mode) noexcept : ha… in CUPMBlasPointerModeGuard()
472 …CUPMBlasPointerModeGuard(const cupmBlasHandle_t &handle, PetscMemType mtype) noexcept : CUPMBlasPo… in CUPMBlasPointerModeGuard()
482 cupmBlasHandle_t handle_;
/petsc/src/ksp/ksp/utils/lmvm/blas_cyclic/cupm/
H A Dblas_cyclic_cupm_impl.hpp22 …static PetscErrorCode axpby_dispatch(cupmBlasHandle_t, cupmBlasInt_t, PetscScalar, const PetscScal…
32 PetscErrorCode BLASCyclic<T>::axpby_dispatch(cupmBlasHandle_t handle, cupmBlasInt_t n, PetscScalar … in axpby_dispatch()
55 cupmBlasHandle_t handle; in axpby()
92 cupmBlasHandle_t handle; in dmv()
134 cupmBlasHandle_t handle; in dsv()
184 cupmBlasHandle_t handle; in trsv()
227 cupmBlasHandle_t handle; in hemv()
270 cupmBlasHandle_t handle; in gemv()
/petsc/src/ksp/ksp/utils/lmvm/dense/cd_cupm/
H A Dcd_cupm.cxx29 cupmBlasHandle_t handle; in SolveInPlace()
53 cupmBlasHandle_t handle; in SolveInPlaceCyclic()
/petsc/src/sys/objects/device/impls/cupm/
H A Dcupmcontext.hpp38 using blas_tag = HandleTag<cupmBlasHandle_t, 1>;
58 cupmBlasHandle_t blas{};
70 PETSC_NODISCARD const cupmBlasHandle_t &get(blas_tag) const noexcept { return this->blas; } in get()
78 static std::array<cupmBlasHandle_t, PETSC_DEVICE_MAX_DEVICES> blashandles_;
602 std::array<typename DeviceContext<T>::cupmBlasHandle_t, PETSC_DEVICE_MAX_DEVICES> DeviceContext<T>:…
/petsc/src/vec/vec/impls/seq/cupm/
H A Dvecseqcupm_impl.hpp468 cupmBlasHandle_t cupmBlasHandle; in AYPXAsync()
511 cupmBlasHandle_t cupmBlasHandle; in AXPYAsync()
819 cupmBlasHandle_t cupmBlasHandle; in WAXPYAsync()
1017 cupmBlasHandle_t cupmBlasHandle; in Dot()
1283 cupmBlasHandle_t handle; in MDot_()
1367 cupmBlasHandle_t cupmBlasHandle; in ScaleAsync()
1404 cupmBlasHandle_t cupmBlasHandle; in TDot()
1516 cupmBlasHandle_t cupmBlasHandle; in SwapAsync()
1569 cupmBlasHandle_t cupmBlasHandle; in AXPBYAsync()
1636 cupmBlasHandle_t cupmBlasHandle; in Norm()
/petsc/src/mat/impls/dense/seq/cupm/
H A Dmatseqdensecupm.hpp610 cupmBlasHandle_t blas_handle; in Solve()
811 cupmBlasHandle_t handle; in MatMultAddColumnRange_Dispatch_()
1300 cupmBlasHandle_t handle; in MatMatMult_Numeric_Dispatch()
1543 cupmBlasHandle_t handle; in Scale()
1592 cupmBlasHandle_t handle; in AXPY()