Searched refs:blas_handle (Results 1 – 2 of 2) sorted by relevance
| /petsc/include/petsc/private/ |
| H A D | cupmobject.hpp | 247 …<T>::GetFromHandleDispatch_(PetscDeviceContext dctx, cupmBlasHandle_t *blas_handle, cupmSolverHand… in GetFromHandleDispatch_() argument
251 if (blas_handle) { in GetFromHandleDispatch_()
252 PetscAssertPointer(blas_handle, 2); in GetFromHandleDispatch_()
253 *blas_handle = nullptr; in GetFromHandleDispatch_()
269 if (blas_handle) PetscCall(PetscDeviceContextGetBLASHandle_Internal(dctx, blas_handle)); in GetFromHandleDispatch_()
281 …ect<T>::GetHandleDispatch_(PetscDeviceContext *dctx, cupmBlasHandle_t *blas_handle, cupmSolverHand… in GetHandleDispatch_() argument
289 PetscCall(GetFromHandleDispatch_(dctx_loc, blas_handle, solver_handle, stream)); in GetHandleDispatch_()
295 …CUPMObject<T>::GetHandles_(PetscDeviceContext *dctx, cupmBlasHandle_t *blas_handle, cupmSolverHand… in GetHandles_() argument
297 return GetHandleDispatch_(dctx, blas_handle, solver_handle, stream); in GetHandles_()
301 …CUPMObject<T>::GetHandles_(PetscDeviceContext *dctx, cupmBlasHandle_t *blas_handle, cupmStream_t *… in GetHandles_() argument
[all …]
|
| /petsc/src/mat/impls/dense/seq/cupm/ |
| H A D | matseqdensecupm.hpp | 610 cupmBlasHandle_t blas_handle; in Solve() local
613 PetscCall(GetHandlesFrom_(dctx, &blas_handle, &solver_handle)); in Solve()
622 …PetscCallCUPMBLAS(cupmBlasXtrsm(blas_handle, CUPMBLAS_SIDE_LEFT, CUPMBLAS_FILL_MODE_UPPER, CUPMBLA… in Solve()
630 …PetscCallCUPMBLAS(cupmBlasXtrsm(blas_handle, CUPMBLAS_SIDE_LEFT, CUPMBLAS_FILL_MODE_UPPER, CUPMBLA… in Solve()
|