Home
last modified time | relevance | path

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

/petsc/include/petsc/private/
H A Dcupmsolverinterface.hpp87 using cupmBlasInt_t = typename BlasInterface<DeviceType::CUDA>::cupmBlasInt_t; typedef
101 …cupmBlasInt_t /* n */, cupmBlasInt_t /* nrhs */, cupmScalar_t * /* A */, cupmBlasInt_t /* lda */, … in cupmSolverXpotrs_bufferSize()
109 …cupmBlasInt_t n, cupmBlasInt_t nrhs, const cupmScalar_t *A, cupmBlasInt_t lda, cupmScalar_t *B, cu… in PETSC_CUPMSOLVER_ALIAS_BLAS_FUNCTION()
128 …cupmBlasInt_t m, cupmBlasInt_t n, cupmScalar_t *A, cupmBlasInt_t lda, cupmScalar_t *work, cupmBlas… in PETSC_CUPMSOLVER_ALIAS_BLAS_FUNCTION()
144 …cupmBlasInt_t /* n */, cupmBlasInt_t /* nrhs */, cupmScalar_t * /* A */, cupmBlasInt_t /* lda */, … in cupmSolverXgetrs_bufferSize()
152 …cupmBlasInt_t n, cupmBlasInt_t nrhs, cupmScalar_t *A, cupmBlasInt_t lda, cupmBlasInt_t *dev_ipiv, … in PETSC_CUPMSOLVER_ALIAS_BLAS_FUNCTION()
H A Dcupmblasinterface.hpp238 using cupmBlasInt_t = int; typedef
310 using cupmBlasInt_t = int; // rocblas will have its own typedef
378 …using cupmBlasInt_t = typename ::Petsc::device::cupm::impl::BlasInterfaceImpl<T>::cupmBlas…
444 …PetscCheck((std::is_same<PetscInt, cupmBlasInt_t>::value) || (x <= std::numeric_limits<cupmBlasInt… in checkCupmBlasIntCast()
449 static PetscErrorCode PetscCUPMBlasIntCast(PetscInt x, cupmBlasInt_t *y) noexcept in PetscCUPMBlasIntCast()
452 *y = static_cast<cupmBlasInt_t>(x); in PetscCUPMBlasIntCast()
/petsc/src/ksp/ksp/utils/lmvm/blas_cyclic/cupm/
H A Dblas_cyclic_cupm_impl.hpp22 …pby_dispatch(cupmBlasHandle_t, cupmBlasInt_t, PetscScalar, const PetscScalar[], PetscScalar, Petsc…
32 …pmBlasHandle_t handle, cupmBlasInt_t n, PetscScalar alpha, const PetscScalar x[], PetscScalar beta… in axpby_dispatch()
53 cupmBlasInt_t m, i_oldest, i_next, y_stride_; in axpby()
70 cupmBlasInt_t diff = i_next - i_oldest; in axpby()
74 cupmBlasInt_t diff = m - i_oldest; in axpby()
90 cupmBlasInt_t m, i_oldest, i_next; in dmv()
112 cupmBlasInt_t diff = i_next - i_oldest; in dmv()
116 cupmBlasInt_t diff = m - i_oldest; in dmv()
132 cupmBlasInt_t m, i_oldest, i_next; in dsv()
153 cupmBlasInt_t diff = i_next - i_oldest; in dsv()
[all …]
/petsc/src/mat/impls/dense/seq/cupm/
H A Dmatseqdensecupm.hpp41 cupmBlasInt_t *d_fact_ipiv; // device pivots
43 cupmBlasInt_t *d_fact_info; // device info
45 cupmBlasInt_t d_fact_lwork; // size of device workspace
55 static PetscErrorCode CheckCUPMSolverInfo_(const cupmBlasInt_t *, cupmStream_t) noexcept;
308 inline PetscErrorCode MatDense_Seq_CUPM<T>::CheckCUPMSolverInfo_(const cupmBlasInt_t *fact_info, cu… in CheckCUPMSolverInfo_()
312 cupmBlasInt_t info = 0; in CheckCUPMSolverInfo_()
336 cupmBlasInt_t lwork; in ResizeFactLwork()
375 const auto m = static_cast<cupmBlasInt_t>(A->rmap->n); in Factor()
376 const auto n = static_cast<cupmBlasInt_t>(A->cmap->n); in Factor()
388 const auto lda = static_cast<cupmBlasInt_t>(MatIMPLCast(A)->lda); in Factor()
[all …]
/petsc/src/ksp/ksp/utils/lmvm/dense/cd_cupm/
H A Dcd_cupm.cxx28 cupmBlasInt_t n; in SolveInPlace()
51 cupmBlasInt_t n_old, n_new; in SolveInPlaceCyclic()
/petsc/src/vec/vec/impls/seq/cupm/
H A Dvecseqcupm_impl.hpp448 const auto n = static_cast<cupmBlasInt_t>(yin->map->n); in AYPXAsync()
510 const auto n = static_cast<cupmBlasInt_t>(yin->map->n); in AXPYAsync()
818 } else if (const auto n = static_cast<cupmBlasInt_t>(win->map->n)) { in WAXPYAsync()
1015 if (const auto n = static_cast<cupmBlasInt_t>(xin->map->n)) { in Dot()
1270 const auto n = static_cast<cupmBlasInt_t>(xin->map->n); in MDot_()
1366 } else if (const auto n = static_cast<cupmBlasInt_t>(xin->map->n)) { in ScaleAsync()
1402 if (const auto n = static_cast<cupmBlasInt_t>(xin->map->n)) { in TDot()
1515 if (const auto n = static_cast<cupmBlasInt_t>(xin->map->n)) { in SwapAsync()
1558 } else if (const auto n = static_cast<cupmBlasInt_t>(yin->map->n)) { in AXPBYAsync()
1640 if (const auto n = static_cast<cupmBlasInt_t>(xin->map->n)) { in Norm()
[all …]