Lines Matching refs:stream
67 …ed_copy(value_type *dest, const value_type *src, size_type n, const StreamBase<U> *stream) noexcept in uninitialized_copy() argument
70 PetscCall(PetscCUPMMemcpyAsync(dest, src, n, cupmMemcpyHostToHost, stream->get_stream(), true)); in uninitialized_copy()
104 …eviceAllocator<T, P>::allocate(value_type **ptr, size_type n, const StreamBase<U> *stream) noexcept in allocate() argument
107 PetscCall(PetscCUPMMallocAsync(ptr, n, stream->get_stream())); in allocate()
113 …cErrorCode DeviceAllocator<T, P>::deallocate(value_type *ptr, const StreamBase<U> *stream) noexcept in deallocate() argument
116 PetscCallCUPM(cupmFreeAsync(ptr, stream->get_stream())); in deallocate()
122 …ode DeviceAllocator<T, P>::zero(value_type *ptr, size_type n, const StreamBase<U> *stream) noexcept in zero() argument
125 PetscCall(PetscCUPMMemsetAsync(ptr, 0, n, stream->get_stream(), true)); in zero()
131 …ed_copy(value_type *dest, const value_type *src, size_type n, const StreamBase<U> *stream) noexcept in uninitialized_copy() argument
134 …PetscCall(PetscCUPMMemcpyAsync(dest, src, n, cupmMemcpyDeviceToDevice, stream->get_stream(), true)… in uninitialized_copy()
140 …viceAllocator<T, P>::set_canary(value_type *ptr, size_type n, const StreamBase<U> *stream) noexcept in set_canary() argument
147 PetscCallThrust(THRUST_CALL(thrust::fill, stream->get_stream(), xptr, xptr + n, canary)); in set_canary()