Home
last modified time | relevance | path

Searched defs:stream (Results 1 – 25 of 28) sorted by relevance

12

/petsc/src/sys/objects/device/impls/cupm/
H A Dcupmallocator.hpp67 …zed_copy(value_type *dest, const value_type *src, size_type n, const StreamBase<U> *stream) noexce… in uninitialized_copy()
104 …DeviceAllocator<T, P>::allocate(value_type **ptr, size_type n, const StreamBase<U> *stream) noexce… in allocate()
113 …scErrorCode DeviceAllocator<T, P>::deallocate(value_type *ptr, const StreamBase<U> *stream) noexce… in deallocate()
122 …Code DeviceAllocator<T, P>::zero(value_type *ptr, size_type n, const StreamBase<U> *stream) noexce… in zero()
131 …zed_copy(value_type *dest, const value_type *src, size_type n, const StreamBase<U> *stream) noexce… in uninitialized_copy()
140 …eviceAllocator<T, P>::set_canary(value_type *ptr, size_type n, const StreamBase<U> *stream) noexce… in set_canary()
H A Dcupmcontext.hpp50 stream_type stream{}; member
487 const auto &stream = impls_cast_(dctx)->stream; in memAlloc() local
504 const auto &stream = impls_cast_(dctx)->stream; in memFree() local
534 const auto stream = impls_cast_(dctx)->stream.get_stream(); in memCopy() local
H A Dcupmevent.hpp161 inline PetscErrorCode CUPMEvent<T>::record(cupmStream_t stream) noexcept in record()
/petsc/include/petsc/private/
H A Dcupmobject.hpp272 cupmStream_t *stream = nullptr; in GetFromHandleDispatch_() local
281 …x, cupmBlasHandle_t *blas_handle, cupmSolverHandle_t *solver_handle, cupmStream_t *stream) noexcept in GetHandleDispatch_()
295 …x, cupmBlasHandle_t *blas_handle, cupmSolverHandle_t *solver_handle, cupmStream_t *stream) noexcept in GetHandles_()
301 …GetHandles_(PetscDeviceContext *dctx, cupmBlasHandle_t *blas_handle, cupmStream_t *stream) noexcept in GetHandles_()
307 …andles_(PetscDeviceContext *dctx, cupmSolverHandle_t *solver_handle, cupmStream_t *stream) noexcept in GetHandles_()
313 inline PetscErrorCode CUPMObject<T>::GetHandles_(PetscDeviceContext *dctx, cupmStream_t *stream) no… in GetHandles_()
331 inline PetscErrorCode CUPMObject<T>::GetHandles_(cupmStream_t *stream) noexcept in GetHandles_()
337 …x, cupmBlasHandle_t *blas_handle, cupmSolverHandle_t *solver_handle, cupmStream_t *stream) noexcept in GetHandlesFrom_()
343 …lesFrom_(PetscDeviceContext dctx, cupmSolverHandle_t *solver_handle, cupmStream_t *stream) noexcept in GetHandlesFrom_()
349 inline PetscErrorCode CUPMObject<T>::GetHandlesFrom_(PetscDeviceContext dctx, cupmStream_t *stream)… in GetHandlesFrom_()
H A Dveccupmimpl.h305 cupmStream_t stream; in ResetAllocatedDevicePtr_() local
400 cupmStream_t stream; in DeviceAllocateCheck_() local
419 cupmStream_t stream; in CopyToDevice_() local
436 cupmStream_t stream; in CopyToHost_() local
601 cupmStream_t stream; in GetArray() local
1074 cupmStream_t stream; in ResetPreallocationCOO_CUPMBase() local
1113 cupmStream_t stream; in SetPreallocationCOO_CUPMBase() local
H A Dmatdensecupmimpl.h411 cupmStream_t stream; in DiagonalUnaryTransform() local
472 cupmStream_t stream; in GetDiagonal() local
/petsc/src/sys/objects/device/impls/
H A Dsegmentedmempool.hpp201 inline PetscErrorCode MemoryChunk<E>::release(const device::StreamBase<U> *stream) noexcept in release()
223 inline PetscErrorCode MemoryChunk<E>::claim(const device::StreamBase<U> *stream, size_type req_size… in claim()
247 inline bool MemoryChunk<E>::can_claim(const device::StreamBase<U> *stream, size_type req_size, bool… in can_claim() argument
347 PetscErrorCode MemoryBlock<T, A, S>::clear_(const stream_type *stream) noexcept in clear_()
366 …<T, A, S>::MemoryBlock(allocator_type *alloc, size_type s, const device::StreamBase<U> *stream) no… in MemoryBlock()
377 stream_type stream; in ~MemoryBlock() local
394 stream_type stream; in operator =() local
430 …Block<T, A, S>::try_allocate_chunk(size_type req_size, T **ptr, const stream_type *stream, bool *s… in try_allocate_chunk()
514 inline PetscErrorCode MemoryBlock<T, A, S>::try_deallocate_chunk(T **ptr, const stream_type *stream… in try_deallocate_chunk()
724 …Type, AllocType, DefaultChunkSize>::make_block_(size_type size, const stream_type *stream) noexcept in make_block_()
[all …]
/petsc/include/
H A Dpetscdevice_hip.h142 static inline hipsolverStatus_t hipsolverGetStream(hipsolverHandle_t handle, hipStream_t *stream) in hipsolverGetStream()
148 static inline hipsolverStatus_t hipsolverSetStream(hipsolverHandle_t handle, hipStream_t stream) in hipsolverSetStream()
171 static inline hipsolverStatus_t hipsolverGetStream(hipsolverHandle_t handle, hipStream_t *stream) in hipsolverGetStream()
177 static inline hipsolverStatus_t hipsolverSetStream(hipsolverHandle_t handle, hipStream_t stream) in hipsolverSetStream()
/petsc/src/mat/impls/dense/seq/cupm/
H A Dmatseqdensecupm.hpp208 cupmStream_t stream; in SetPreallocation_() local
246 cupmStream_t stream; in HostToDevice_() local
283 cupmStream_t stream; in DeviceToHost_() local
308 …nse_Seq_CUPM<T>::CheckCUPMSolverInfo_(const cupmBlasInt_t *fact_info, cupmStream_t stream) noexcept in CheckCUPMSolverInfo_()
377 cupmStream_t stream; in Factor() local
463 cupmStream_t stream; in Factor() local
563 cupmStream_t stream; in Factor() local
652 cupmStream_t stream; in MatSolve_Factored_Dispatch_() local
727 cupmStream_t stream; in MatMatSolve_Factored_Dispatch_() local
1031 cupmStream_t stream; in Reset() local
[all …]
/petsc/src/vec/vec/impls/seq/cupm/
H A Dvecseqcupm_impl.hpp189 cupmStream_t stream; in PointwiseBinary_() local
237 cupmStream_t stream; in PointwiseUnary_() local
370 cupmStream_t stream; in GetLocalVector() local
430 cupmStream_t stream; in RestoreLocalVector() local
459 cupmStream_t stream; in AYPXAsync() local
820 cupmStream_t stream; in WAXPYAsync() local
907 …de VecSeq_CUPM<T>::MAXPY_kernel_dispatch_(PetscDeviceContext dctx, cupmStream_t stream, PetscScala… in MAXPY_kernel_dispatch_()
924 …de VecSeq_CUPM<T>::MAXPY_kernel_dispatch_(PetscDeviceContext dctx, cupmStream_t stream, PetscScala… in MAXPY_kernel_dispatch_()
937 cupmStream_t stream; in MAXPYAsync() local
1152 …ode VecSeq_CUPM<T>::MDot_kernel_dispatch_(PetscDeviceContext dctx, cupmStream_t stream, const Pets… in MDot_kernel_dispatch_()
[all …]
/petsc/src/mat/impls/hypre/cuda/
H A Dhypre1.cu9 cudaStream_t stream; in MatZeroRows_CUDA() local
/petsc/src/mat/impls/hypre/hip/
H A Dhypre2.hip.cxx9 hipStream_t stream; in MatZeroRows_HIP() local
/petsc/src/sys/objects/device/impls/cupm/cuda/
H A Dcupmcontext.cu39 PetscErrorCode PetscGetCurrentCUDAStream(cudaStream_t *stream) in PetscGetCurrentCUDAStream()
/petsc/src/sys/objects/device/impls/cupm/hip/
H A Dcupmcontext.hip.cxx47 PetscErrorCode PetscGetCurrentHIPStream(hipStream_t *stream) in PetscGetCurrentHIPStream()
/petsc/src/mat/impls/h2opus/
H A Dmath2opussampler.hpp18 h2opusComputeStream_t stream; member in PetscMatrixSampler
67 void PetscMatrixSampler::SetStream(h2opusComputeStream_t stream) in SetStream()
/petsc/src/tao/unconstrained/tutorials/
H A Drosenbrock4.h360 static PetscErrorCode RosenbrockObjective_Device(cupmStream_t stream, Rosenbrock r, const PetscScal… in RosenbrockObjective_Device()
370 static PetscErrorCode RosenbrockGradient_Device(cupmStream_t stream, Rosenbrock r, const PetscScala… in RosenbrockGradient_Device()
380 static PetscErrorCode RosenbrockObjectiveGradient_Device(cupmStream_t stream, Rosenbrock r, const P… in RosenbrockObjectiveGradient_Device()
390 static PetscErrorCode RosenbrockHessian_Device(cupmStream_t stream, Rosenbrock r, const PetscScalar… in RosenbrockHessian_Device()
486 cupmStream_t *stream; in FormObjective() local
524 cupmStream_t *stream; in FormGradient() local
583 cupmStream_t *stream; in FormObjectiveGradient() local
640 cupmStream_t *stream; in FormHessian() local
/petsc/src/ksp/ksp/utils/lmvm/blas_cyclic/cupm/
H A Dblas_cyclic_cupm_impl.hpp135 cupmStream_t stream; in dsv() local
185 cupmStream_t stream; in trsv() local
228 cupmStream_t stream; in hemv() local
271 cupmStream_t stream; in gemv() local
/petsc/src/sys/classes/random/impls/curand/
H A Dcurand.c81 cudaStream_t *stream; in PetscRandomCreate_CURAND() local
/petsc/
H A Dmakefile450 stream: mpistream target
/petsc/lib/petsc/bin/maint/
H A Dgcov.py44 def __log(stream, *args, **kwargs): argument
/petsc/src/vec/vec/impls/mpi/cupm/
H A Dvecmpicupm_impl.hpp322 cupmStream_t stream; in SetValuesCOO() local
/petsc/src/sys/classes/viewer/impls/glvis/
H A Dglvis.c721 FILE *stream; in PetscViewerDestroy_ASCII_Socket() local
/petsc/src/vec/is/sf/impls/basic/cupm/
H A Dsfcupm_impl.hpp784 cupmStream_t *stream; in LinkSetUp() local
/petsc/src/vec/is/sf/impls/basic/
H A Dsfpack.h168 …cupmStream_t stream; /* stream on which input/output root/leafdata is computed o… member
/petsc/src/binding/petsc4py/src/petsc4py/PETSc/
H A DVec.pyx719 def __dlpack__(self, stream=-1): argument

12