Searched refs:PETSC_MEMTYPE_CUPM (Results 1 – 6 of 6) sorted by relevance
| /petsc/src/vec/vec/impls/mpi/cupm/ |
| H A D | vecmpicupm_impl.hpp | 340 PetscCall(PetscDeviceMalloc(dctx, PETSC_MEMTYPE_CUPM(), size, &vv)); in SetValuesCOO()
352 …all(PetscSFReduceWithMemTypeBegin(sf, MPIU_SCALAR, PETSC_MEMTYPE_CUPM(), sendbuf_d, PETSC_MEMTYPE_… in SetValuesCOO()
|
| /petsc/include/petsc/private/ |
| H A D | cupminterface.hpp | 118 …PETSC_NODISCARD static constexpr PetscMemType PETSC_MEMTYPE_CUPM() noexcept { return T == DeviceTy… in PETSC_MEMTYPE_CUPM() function
130 using ::Petsc::device::cupm::impl::InterfaceBase<T>::PETSC_MEMTYPE_CUPM
599 …ype = ((cerr == cupmSuccess) && (mtype == cupmMemoryTypeDevice)) ? PETSC_MEMTYPE_CUPM() : PETSC_ME… in PetscCUPMGetMemType()
|
| H A D | veccupmimpl.h | 673 …scDefined(HAVE_NVSHMEM) && VecCUPMCast(v)->nvshmem) ? PETSC_MEMTYPE_NVSHMEM : PETSC_MEMTYPE_CUPM(); in GetArrayAndMemtype()
902 PetscCall(CheckPointerMatchesMemType_(device_array, PETSC_MEMTYPE_CUPM())); in Initialize_CUPMBase()
|
| /petsc/src/vec/vec/impls/seq/cupm/ |
| H A D | vecseqcupm_impl.hpp | 956 PetscCall(PetscDeviceMalloc(dctx, PETSC_MEMTYPE_CUPM(), nv, &d_alpha)); in MAXPYAsync()
1196 PetscCall(PetscDeviceMalloc(dctx, PETSC_MEMTYPE_CUPM(), nwork, &d_results)); in MDot_()
1278 PetscCall(PetscDeviceMalloc(dctx, PETSC_MEMTYPE_CUPM(), nv, &d_z)); in MDot_()
2261 PetscCall(PetscDeviceMalloc(dctx, PETSC_MEMTYPE_CUPM(), size, &vv)); in SetValuesCOO()
|
| /petsc/src/sys/objects/device/impls/cupm/ |
| H A D | cupmcontext.hpp | 191 …MemTypeHost(mtype) || (mtype == PETSC_MEMTYPE_DEVICE) || (mtype == PETSC_MEMTYPE_CUPM()), PETSC_CO… in check_memtype_()
|
| /petsc/src/mat/impls/dense/seq/cupm/ |
| H A D | matseqdensecupm.hpp | 1214 if (mtype) *mtype = PETSC_MEMTYPE_CUPM(); in GetArrayAndMemType()
|