Searched refs:DeviceAllocator (Results 1 – 2 of 2) sorted by relevance
| /petsc/src/sys/objects/device/impls/cupm/ |
| H A D | cupmallocator.hpp | 79 class DeviceAllocator;
82 class PETSC_SINGLE_LIBRARY_VISIBILITY_INTERNAL DeviceAllocator : public memory::impl::SegmentedMemo… class
104 inline PetscErrorCode DeviceAllocator<T, P>::allocate(value_type **ptr, size_type n, const StreamBa… in allocate()
113 inline PetscErrorCode DeviceAllocator<T, P>::deallocate(value_type *ptr, const StreamBase<U> *strea… in deallocate()
122 inline PetscErrorCode DeviceAllocator<T, P>::zero(value_type *ptr, size_type n, const StreamBase<U>… in zero()
131 inline PetscErrorCode DeviceAllocator<T, P>::uninitialized_copy(value_type *dest, const value_type … in uninitialized_copy()
140 inline PetscErrorCode DeviceAllocator<T, P>::set_canary(value_type *ptr, size_type n, const StreamB… in set_canary()
|
| H A D | cupmcontext.hpp | 495 …PetscCall(default_pool_<DeviceAllocator<T>>().allocate(n, reinterpret_cast<char **>(dest), &stream… in memAlloc()
524 …PetscCall(default_pool_<DeviceAllocator<T>>().deallocate(reinterpret_cast<char **>(ptr), &stream)); in memFree()
|