Searched refs:usehost (Results 1 – 4 of 4) sorted by relevance
| /petsc/src/mat/impls/dense/mpi/cupm/ |
| H A D | matmpidensecupm.hpp | 241 inline PetscErrorCode MatDense_MPI_CUPM<T>::BindToCPU(Mat A, PetscBool usehost) noexcept in BindToCPU() argument
249 if (const auto mimpl_A = mimpl->A) PetscCall(MatBindToCPU(mimpl_A, usehost)); in BindToCPU()
250 A->boundtocpu = usehost; in BindToCPU()
251 PetscCall(PetscStrFreeAllocpy(usehost ? PETSCRANDER48 : PETSCDEVICERAND(), &A->defaultrandtype)); in BindToCPU()
252 if (!usehost) { in BindToCPU()
261 …MatComposeOp_CUPM(usehost, pobj, "MatDenseGetColumnVec_C", MatDenseGetColumnVec_MPIDense, GetColum… in BindToCPU()
262 …MatComposeOp_CUPM(usehost, pobj, "MatDenseRestoreColumnVec_C", MatDenseRestoreColumnVec_MPIDense, … in BindToCPU()
263 …MatComposeOp_CUPM(usehost, pobj, "MatDenseGetColumnVecRead_C", MatDenseGetColumnVecRead_MPIDense, … in BindToCPU()
264 …MatComposeOp_CUPM(usehost, pobj, "MatDenseRestoreColumnVecRead_C", MatDenseRestoreColumnVecRead_MP… in BindToCPU()
265 …MatComposeOp_CUPM(usehost, pobj, "MatDenseGetColumnVecWrite_C", MatDenseGetColumnVecWrite_MPIDense… in BindToCPU()
[all …]
|
| /petsc/src/vec/vec/impls/mpi/cupm/ |
| H A D | vecmpicupm_impl.hpp | 145 inline PetscErrorCode VecMPI_CUPM<T>::BindToCPU(Vec v, PetscBool usehost) noexcept in BindToCPU() argument
151 PetscCall(BindToCPU_CUPMBase(v, usehost, dctx)); in BindToCPU()
|
| /petsc/include/petsc/private/ |
| H A D | veccupmimpl.h | 968 if (usehost) { \
977 inline PetscErrorCode Vec_CUPMBase<T, D>::BindToCPU_CUPMBase(Vec v, PetscBool usehost, PetscDeviceC… in BindToCPU_CUPMBase() argument
980 v->boundtocpu = usehost; in BindToCPU_CUPMBase()
981 if (usehost) PetscCall(CopyToHost_(dctx, v)); in BindToCPU_CUPMBase()
982 PetscCall(PetscStrFreeAllocpy(usehost ? PETSCRANDER48 : PETSCDEVICERAND(), &v->defaultrandtype)); in BindToCPU_CUPMBase()
|
| /petsc/src/vec/vec/impls/seq/cupm/ |
| H A D | vecseqcupm_impl.hpp | 321 inline PetscErrorCode VecSeq_CUPM<T>::BindToCPU(Vec v, PetscBool usehost) noexcept in BindToCPU() argument
327 PetscCall(BindToCPU_CUPMBase(v, usehost, dctx)); in BindToCPU()
|