| /petsc/src/vec/vec/impls/mpi/cupm/cuda/ |
| H A D | vecmpicupm.cu | 57 PetscErrorCode VecCUDAGetArrays_Private(Vec v, const PetscScalar **host_array, const PetscScalar **… in VecCUDAGetArrays_Private() argument
64 PetscCall(VecMPI_CUDA.GetArrays_CUPMBase(v, host_array, device_array, mask, dctx)); in VecCUDAGetArrays_Private()
|
| /petsc/src/vec/vec/impls/mpi/cupm/hip/ |
| H A D | vecmpicupm.hip.cxx | 57 PetscErrorCode VecHIPGetArrays_Private(Vec v, const PetscScalar **host_array, const PetscScalar **d… in VecHIPGetArrays_Private() argument
64 PetscCall(VecMPI_HIP.GetArrays_CUPMBase(v, host_array, device_array, mask, dctx)); in VecHIPGetArrays_Private()
|
| /petsc/include/petsc/private/ |
| H A D | veccupmimpl.h | 520 …te_IMPL_Private(Vec v, PetscBool *alloc_missing, PetscInt nghost, PetscScalar *host_array) noexcept in VecCreate_IMPL_Private() argument
522 return D::VecCreate_IMPL_Private_(v, alloc_missing, nghost, host_array); in VecCreate_IMPL_Private()
761 auto &host_array = vimpl->array_allocated; in ReplaceArray() local
765 if (host_array != vimpl->array) PetscCall(CopyToHost_(dctx, v)); in ReplaceArray()
766 if (host_array) { in ReplaceArray()
769 PetscCall(PetscFree(host_array)); in ReplaceArray()
771 host_array = aptr; in ReplaceArray()
772 vimpl->array = host_array; in ReplaceArray()
815 auto &host_array = vimpl->unplacedarray; in ResetArray() local
817 PetscCall(CheckPointerMatchesMemType_(host_array, PETSC_MEMTYPE_DEVICE)); in ResetArray()
[all …]
|
| /petsc/src/vec/vec/impls/mpi/cupm/ |
| H A D | vecmpicupm_impl.hpp | 74 …ontext dctx, PetscBool allocate_missing, PetscInt nghost, PetscScalar *host_array, PetscScalar *de… in CreateMPICUPM_() argument
78 PetscCall(Initialize_CUPMBase(v, allocate_missing, host_array, device_array, dctx)); in CreateMPICUPM_()
102 …(MPI_Comm comm, PetscInt bs, PetscInt n, PetscInt N, const PetscScalar host_array[], const PetscSc… in CreateMPICUPMWithArrays() argument
111 …PetscCall(CreateMPICUPM_(*v, dctx, PETSC_FALSE, 0, PetscRemoveConstCast(host_array), PetscRemoveCo… in CreateMPICUPMWithArrays()
|
| /petsc/src/vec/vec/impls/seq/cupm/ |
| H A D | vecseqcupm_impl.hpp | 137 …ecCreate_IMPL_Private_(Vec v, PetscBool *alloc_missing, PetscInt, PetscScalar *host_array) noexcept in VecCreate_IMPL_Private_() argument
145 PetscCall(VecCreate_Seq_Private(v, host_array)); in VecCreate_IMPL_Private_()
175 …q_CUPM<T>::CreateSeqCUPM_(Vec v, PetscDeviceContext dctx, PetscScalar *host_array, PetscScalar *de… in CreateSeqCUPM_() argument
178 PetscCall(base_type::VecCreate_IMPL_Private(v, nullptr, 0, host_array)); in CreateSeqCUPM_()
179 PetscCall(Initialize_CUPMBase(v, PETSC_FALSE, host_array, device_array, dctx)); in CreateSeqCUPM_()
290 …thBothArrays(MPI_Comm comm, PetscInt bs, PetscInt n, const PetscScalar host_array[], const PetscSc… in CreateSeqCUPMWithBothArrays() argument
299 …PetscCall(CreateSeqCUPM_(*v, dctx, PetscRemoveConstCast(host_array), PetscRemoveConstCast(device_a… in CreateSeqCUPMWithBothArrays()
|
| /petsc/src/mat/impls/dense/seq/cupm/ |
| H A D | matseqdensecupm.hpp | 1862 if (const auto host_array = mimpl->v) { in GetSubMatrix() local
1864 PetscCall(MatDensePlaceArray(cmat, array_offset(host_array))); in GetSubMatrix()
|