Searched defs:device_array (Results 1 – 10 of 10) sorted by relevance
| /petsc/src/vec/vec/impls/mpi/cupm/hip/ |
| H A D | vecmpicupm.hip.cxx | 57 …HIPGetArrays_Private(Vec v, const PetscScalar **host_array, const PetscScalar **device_array, Pets… in VecHIPGetArrays_Private()
|
| /petsc/src/vec/vec/impls/mpi/cupm/cuda/ |
| H A D | vecmpicupm.cu | 57 …UDAGetArrays_Private(Vec v, const PetscScalar **host_array, const PetscScalar **device_array, Pets… in VecCUDAGetArrays_Private()
|
| /petsc/src/mat/impls/dense/mpi/cupm/cuda/ |
| H A D | matmpidensecuda.cu | 334 PetscErrorCode MatDenseCUDASetPreallocation(Mat A, PetscScalar *device_array) in MatDenseCUDASetPreallocation()
|
| /petsc/src/mat/impls/dense/mpi/cupm/hip/ |
| H A D | matmpidensehip.hip.cxx | 335 PetscErrorCode MatDenseHIPSetPreallocation(Mat A, PetscScalar *device_array) in MatDenseHIPSetPreallocation()
|
| /petsc/src/vec/vec/impls/mpi/cupm/ |
| H A D | vecmpicupm_impl.hpp | 74 …ool allocate_missing, PetscInt nghost, PetscScalar *host_array, PetscScalar *device_array) noexcept in CreateMPICUPM_()
102 …bs, PetscInt n, PetscInt N, const PetscScalar host_array[], const PetscScalar device_array[], Vec … in CreateMPICUPMWithArrays()
|
| /petsc/include/petsc/private/ |
| H A D | veccupmimpl.h | 298 auto &device_array = VecCUPMCast(v)->array_allocated_d; in ResetAllocatedDevicePtr_() local
894 …ase(Vec v, PetscBool allocate_missing, PetscScalar *host_array, PetscScalar *device_array, PetscDe… in Initialize_CUPMBase()
1053 …::GetArrays_CUPMBase(Vec v, const PetscScalar **host_array, const PetscScalar **device_array, Pets… in GetArrays_CUPMBase()
|
| H A D | matdensecupmimpl.h | 288 …se_CUPM<T, D>::SetPreallocation(Mat A, PetscDeviceContext dctx, PetscScalar *device_array) noexcept in SetPreallocation()
|
| /petsc/src/mat/impls/dense/mpi/cupm/ |
| H A D | matmpidensecupm.hpp | 117 …_MPI_CUPM<T>::SetPreallocation_(Mat A, PetscDeviceContext dctx, PetscScalar *device_array) noexcept in SetPreallocation_()
|
| /petsc/src/vec/vec/impls/seq/cupm/ |
| H A D | vecseqcupm_impl.hpp | 175 …qCUPM_(Vec v, PetscDeviceContext dctx, PetscScalar *host_array, PetscScalar *device_array) noexcept in CreateSeqCUPM_()
290 …m, PetscInt bs, PetscInt n, const PetscScalar host_array[], const PetscScalar device_array[], Vec … in CreateSeqCUPMWithBothArrays()
369 if (auto &device_array = wcu->array_d) { in GetLocalVector() local
|
| /petsc/src/mat/impls/dense/seq/cupm/ |
| H A D | matseqdensecupm.hpp | 1852 const auto device_array = array_offset(MatCUPMCast(A)->d_v); in GetSubMatrix() local
|