Lines Matching refs:ptr
49 inline PetscErrorCode HostAllocator<T, P>::allocate(value_type **ptr, size_type n, const StreamBase… in allocate() argument
52 PetscCall(PetscCUPMMallocHost(ptr, n)); in allocate()
58 inline PetscErrorCode HostAllocator<T, P>::deallocate(value_type *ptr, const StreamBase<U> *) noexc… in deallocate() argument
61 PetscCallCUPM(cupmFreeHost(ptr)); in deallocate()
104 inline PetscErrorCode DeviceAllocator<T, P>::allocate(value_type **ptr, size_type n, const StreamBa… in allocate() argument
107 PetscCall(PetscCUPMMallocAsync(ptr, n, stream->get_stream())); in allocate()
113 inline PetscErrorCode DeviceAllocator<T, P>::deallocate(value_type *ptr, const StreamBase<U> *strea… in deallocate() argument
116 PetscCallCUPM(cupmFreeAsync(ptr, stream->get_stream())); in deallocate()
122 inline PetscErrorCode DeviceAllocator<T, P>::zero(value_type *ptr, size_type n, const StreamBase<U>… in zero() argument
125 PetscCall(PetscCUPMMemsetAsync(ptr, 0, n, stream->get_stream(), true)); in zero()
140 inline PetscErrorCode DeviceAllocator<T, P>::set_canary(value_type *ptr, size_type n, const StreamB… in set_canary() argument
144 const auto xptr = thrust::device_pointer_cast(ptr); in set_canary()