Home
last modified time | relevance | path

Searched defs:dctx (Results 1 – 25 of 56) sorted by relevance

123

/petsc/src/sys/objects/device/interface/
H A Ddcontext.cxx26 static PetscErrorCode destroy_(PetscDeviceContext dctx) noexcept in destroy_()
38 static PetscErrorCode reset_(PetscDeviceContext dctx, bool zero = true) noexcept in reset_()
61 static PetscErrorCode invalidate_(PetscDeviceContext dctx) noexcept in invalidate_()
101 PetscErrorCode PetscDeviceContextCreate(PetscDeviceContext *dctx) in PetscDeviceContextCreate()
146 PetscErrorCode PetscDeviceContextDestroy(PetscDeviceContext *dctx) in PetscDeviceContextDestroy()
181 PetscErrorCode PetscDeviceContextSetStreamType(PetscDeviceContext dctx, PetscStreamType type) in PetscDeviceContextSetStreamType()
220 PetscErrorCode PetscDeviceContextGetStreamType(PetscDeviceContext dctx, PetscStreamType *type) in PetscDeviceContextGetStreamType()
242 static PetscErrorCode PetscDeviceContextSetDevice_Private(PetscDeviceContext dctx, PetscDevice devi… in PetscDeviceContextSetDevice_Private()
263 PetscErrorCode PetscDeviceContextSetDefaultDeviceForType_Internal(PetscDeviceContext dctx, PetscDev… in PetscDeviceContextSetDefaultDeviceForType_Internal()
298 PetscErrorCode PetscDeviceContextSetDevice(PetscDeviceContext dctx, PetscDevice device) in PetscDeviceContextSetDevice()
[all …]
H A Dmark_dcontext.cxx56 static PetscErrorCode PetscDeviceContextCreateEvent_Private(PetscDeviceContext dctx, PetscEvent *ev… in PetscDeviceContextCreateEvent_Private()
75 static PetscErrorCode PetscDeviceContextRecordEvent_Private(PetscDeviceContext dctx, PetscEvent eve… in PetscDeviceContextRecordEvent_Private()
105 static PetscErrorCode PetscDeviceContextWaitForEvent_Private(PetscDeviceContext dctx, PetscEvent ev… in PetscDeviceContextWaitForEvent_Private()
299 inline PetscEvent MarkedObjectMap::snapshot_type::init_event_(PetscDeviceContext dctx) noexcept in init_event_()
313 MarkedObjectMap::snapshot_type::snapshot_type(PetscDeviceContext dctx, frame_type frame) noexcept :… in snapshot_type()
336 PetscErrorCode MarkedObjectMap::snapshot_type::ensure_event(PetscDeviceContext dctx) noexcept in ensure_event()
391 static PetscErrorCode PetscDeviceContextMapIterVisitor(PetscDeviceContext dctx, T &&callback) noexc… in PetscDeviceContextMapIterVisitor()
421 PetscErrorCode PetscDeviceContextSyncClearMap_Internal(PetscDeviceContext dctx) in PetscDeviceContextSyncClearMap_Internal()
469 PetscErrorCode PetscDeviceContextCheckNotOrphaned_Internal(PetscDeviceContext dctx) in PetscDeviceContextCheckNotOrphaned_Internal()
500 …MarkFromID_CompatibleModes(MarkedObjectMap::mapped_type &marked, PetscDeviceContext dctx, PetscObj… in MarkFromID_CompatibleModes()
[all …]
H A Dglobal_dcontext.cxx26 static inline PetscErrorCode PetscSetDefaultCUPMStreamFromDeviceContext(PetscDeviceContext dctx, Pe… in PetscSetDefaultCUPMStreamFromDeviceContext()
107 PetscErrorCode PetscDeviceContextGetCurrentContext(PetscDeviceContext *dctx) in PetscDeviceContextGetCurrentContext()
142 PetscErrorCode PetscDeviceContextSetCurrentContext(PetscDeviceContext dctx) in PetscDeviceContextSetCurrentContext()
H A Dpetscdevice_interface_internal.hpp30 #define PetscDebugInfo(dctx, ...) PetscInfo(dctx, __VA_ARGS__) argument
32 #define PetscDebugInfo(dctx, ...) PETSC_SUCCESS argument
88 explicit CxxData(PetscDeviceContext dctx) noexcept : self_{dctx, NoOpDeleter{}} { } in CxxData()
118 inline PetscErrorCode CxxData::reset_self(PetscDeviceContext dctx) noexcept in reset_self()
169 PETSC_NODISCARD inline CxxData *CxxDataCast(PetscDeviceContext dctx) noexcept in CxxDataCast()
H A Dmemory.cxx180 static PetscErrorCode PetscDeviceCheckCapable_Private(PetscDeviceContext dctx, bool cond, const cha… in PetscDeviceCheckCapable_Private()
325 PetscErrorCode PetscDeviceAllocate_Private(PetscDeviceContext dctx, PetscBool clear, PetscMemType m… in PetscDeviceAllocate_Private()
391 PetscErrorCode PetscDeviceDeallocate_Private(PetscDeviceContext dctx, void *PETSC_RESTRICT ptr) in PetscDeviceDeallocate_Private()
470 PetscErrorCode PetscDeviceMemcpy(PetscDeviceContext dctx, void *PETSC_RESTRICT dest, const void *PE… in PetscDeviceMemcpy()
543 PetscErrorCode PetscDeviceMemset(PetscDeviceContext dctx, void *ptr, PetscInt v, std::size_t n) in PetscDeviceMemset()
/petsc/src/sys/objects/device/c/
H A Ddevice.c74 PetscErrorCode PetscDeviceContextCreate(PetscDeviceContext *dctx) in PetscDeviceContextCreate()
82 PetscErrorCode PetscDeviceContextDestroy(PetscDeviceContext *dctx) in PetscDeviceContextDestroy()
90 PetscErrorCode PetscDeviceContextSetStreamType(PETSC_UNUSED PetscDeviceContext dctx, PETSC_UNUSED P… in PetscDeviceContextSetStreamType()
96 PetscErrorCode PetscDeviceContextGetStreamType(PETSC_UNUSED PetscDeviceContext dctx, PetscStreamTyp… in PetscDeviceContextGetStreamType()
104 PetscErrorCode PetscDeviceContextSetDevice(PETSC_UNUSED PetscDeviceContext dctx, PETSC_UNUSED Petsc… in PetscDeviceContextSetDevice()
110 PetscErrorCode PetscDeviceContextGetDevice(PETSC_UNUSED PetscDeviceContext dctx, PetscDevice *devic… in PetscDeviceContextGetDevice()
118 PetscErrorCode PetscDeviceContextGetDeviceType(PETSC_UNUSED PetscDeviceContext dctx, PetscDeviceTyp… in PetscDeviceContextGetDeviceType()
126 PetscErrorCode PetscDeviceContextSetUp(PETSC_UNUSED PetscDeviceContext dctx) in PetscDeviceContextSetUp()
132 PetscErrorCode PetscDeviceContextDuplicate(PETSC_UNUSED PetscDeviceContext dctx, PetscDeviceContext… in PetscDeviceContextDuplicate()
140 PetscErrorCode PetscDeviceContextQueryIdle(PETSC_UNUSED PetscDeviceContext dctx, PetscBool *idle) in PetscDeviceContextQueryIdle()
[all …]
/petsc/src/sys/objects/device/impls/sycl/
H A Dsyclcontext.sycl.cxx41 static PetscErrorCode initialize_(PetscInt id, PetscDeviceContext dctx) noexcept in initialize_()
60 static PetscErrorCode destroy(PetscDeviceContext dctx) noexcept in destroy()
68 static PetscErrorCode setUp(PetscDeviceContext dctx) noexcept in setUp()
85 static PetscErrorCode query(PetscDeviceContext dctx, PetscBool *idle) noexcept in query()
94 static PetscErrorCode synchronize(PetscDeviceContext dctx) noexcept in synchronize()
105 static PetscErrorCode getStreamHandle(PetscDeviceContext dctx, void **handle) noexcept in getStreamHandle()
112 static PetscErrorCode beginTimer(PetscDeviceContext dctx) noexcept in beginTimer()
140 static PetscErrorCode endTimer(PetscDeviceContext dctx, PetscLogDouble *elapsed) noexcept in endTimer()
170 PetscErrorCode PetscDeviceContextCreate_SYCL(PetscDeviceContext dctx) in PetscDeviceContextCreate_SYCL()
/petsc/src/sys/objects/device/impls/cupm/hip/
H A Dcupmcontext.hip.cxx5 PetscErrorCode PetscDeviceContextCreate_HIP(PetscDeviceContext dctx) in PetscDeviceContextCreate_HIP()
27 PetscDeviceContext dctx; in PetscHIPBLASGetHandle() local
38 PetscDeviceContext dctx; in PetscHIPSOLVERGetHandle() local
49 PetscDeviceContext dctx; in PetscGetCurrentHIPStream() local
/petsc/src/sys/objects/device/impls/cupm/cuda/
H A Dcupmcontext.cu5 PetscErrorCode PetscDeviceContextCreate_CUDA(PetscDeviceContext dctx) in PetscDeviceContextCreate_CUDA()
19 PetscDeviceContext dctx; in PetscCUBLASGetHandle() local
30 PetscDeviceContext dctx; in PetscCUSOLVERDnGetHandle() local
41 PetscDeviceContext dctx; in PetscGetCurrentCUDAStream() local
/petsc/src/sys/objects/device/impls/cupm/
H A Dcupmcontext.hpp93 static PetscErrorCode initialize_handle_(blas_tag, PetscDeviceContext dctx) noexcept in initialize_handle_()
120 static PetscErrorCode initialize_handle_(solver_tag, PetscDeviceContext dctx) noexcept in initialize_handle_()
160 …static PetscErrorCode check_current_device_(PetscDeviceContext dctx) noexcept { return check_curre… in check_current_device_()
280 inline PetscErrorCode DeviceContext<T>::destroy(PetscDeviceContext dctx) noexcept in destroy()
295 inline PetscErrorCode DeviceContext<T>::changeStreamType(PetscDeviceContext dctx, PETSC_UNUSED Pets… in changeStreamType()
308 inline PetscErrorCode DeviceContext<T>::setUp(PetscDeviceContext dctx) noexcept in setUp()
324 inline PetscErrorCode DeviceContext<T>::query(PetscDeviceContext dctx, PetscBool *idle) noexcept in query()
359 inline PetscErrorCode DeviceContext<T>::synchronize(PetscDeviceContext dctx) noexcept in synchronize()
371 inline PetscErrorCode DeviceContext<T>::getHandle(PetscDeviceContext dctx, void *handle) noexcept in getHandle()
381 inline PetscErrorCode DeviceContext<T>::getHandlePtr(PetscDeviceContext dctx, void **handle) noexce… in getHandlePtr()
[all …]
/petsc/include/
H A Dpetscdevice.h144 #define PetscDeviceMalloc(dctx, mtype, n, ptr) PetscDeviceAllocate_Private((dctx), PETSC_FALSE, (mt… argument
178 #define PetscDeviceCalloc(dctx, mtype, n, ptr) PetscDeviceAllocate_Private((dctx), PETSC_TRUE, (mty… argument
215 #define PetscDeviceFree(dctx, ptr) ((PetscErrorCode)(PetscDeviceDeallocate_Private((dctx), (ptr)) |… argument
259 #define PetscDeviceArrayCopy(dctx, dest, src, n) PetscDeviceMemcpy((dctx), (dest), (src), (size_t)(… argument
300 #define PetscDeviceArrayZero(dctx, ptr, n) PetscDeviceMemset((dctx), (ptr), 0, (size_t)(n) * sizeof… argument
/petsc/include/petsc/private/
H A Dveccupmimpl.h275 inline Vec_CUPMBase<T, D>::VectorArray<MT, MA>::VectorArray(PetscDeviceContext dctx, Vec v) noexcep… in VectorArray()
296 inline PetscErrorCode Vec_CUPMBase<T, D>::ResetAllocatedDevicePtr_(PetscDeviceContext dctx, Vec v, … in ResetAllocatedDevicePtr_()
392 inline PetscErrorCode Vec_CUPMBase<T, D>::DeviceAllocateCheck_(PetscDeviceContext dctx, Vec v) noex… in DeviceAllocateCheck_()
414 inline PetscErrorCode Vec_CUPMBase<T, D>::CopyToDevice_(PetscDeviceContext dctx, Vec v, bool forcea… in CopyToDevice_()
431 inline PetscErrorCode Vec_CUPMBase<T, D>::CopyToHost_(PetscDeviceContext dctx, Vec v, bool forceasy… in CopyToHost_()
452 PetscDeviceContext dctx; in DestroyDevice_() local
572 inline PetscErrorCode Vec_CUPMBase<T, D>::GetArray(Vec v, PetscScalar **a, PetscDeviceContext dctx)… in GetArray()
626 PetscDeviceContext dctx; in GetArray() local
659 PetscDeviceContext dctx; in RestoreArray() local
669 …::GetArrayAndMemtype(Vec v, PetscScalar **a, PetscMemType *mtype, PetscDeviceContext dctx) noexcept in GetArrayAndMemtype()
[all …]
H A Dcupmobject.hpp160 constexpr inline RestoreableArray<T, MT, MA>::RestoreableArray(PetscDeviceContext dctx) noexcept : … in RestoreableArray()
247 inline PetscErrorCode CUPMObject<T>::GetFromHandleDispatch_(PetscDeviceContext dctx, cupmBlasHandle… in GetFromHandleDispatch_()
281 inline PetscErrorCode CUPMObject<T>::GetHandleDispatch_(PetscDeviceContext *dctx, cupmBlasHandle_t … in GetHandleDispatch_()
295 inline PetscErrorCode CUPMObject<T>::GetHandles_(PetscDeviceContext *dctx, cupmBlasHandle_t *blas_h… in GetHandles_()
301 inline PetscErrorCode CUPMObject<T>::GetHandles_(PetscDeviceContext *dctx, cupmBlasHandle_t *blas_h… in GetHandles_()
307 inline PetscErrorCode CUPMObject<T>::GetHandles_(PetscDeviceContext *dctx, cupmSolverHandle_t *solv… in GetHandles_()
313 inline PetscErrorCode CUPMObject<T>::GetHandles_(PetscDeviceContext *dctx, cupmStream_t *stream) no… in GetHandles_()
337 inline PetscErrorCode CUPMObject<T>::GetHandlesFrom_(PetscDeviceContext dctx, cupmBlasHandle_t *bla… in GetHandlesFrom_()
343 inline PetscErrorCode CUPMObject<T>::GetHandlesFrom_(PetscDeviceContext dctx, cupmSolverHandle_t *s… in GetHandlesFrom_()
349 inline PetscErrorCode CUPMObject<T>::GetHandlesFrom_(PetscDeviceContext dctx, cupmStream_t *stream)… in GetHandlesFrom_()
H A Ddeviceimpl.h120 #define PetscValidDeviceContext(dctx, argno) \ argument
154 #define PetscValidDeviceContext(dctx, argno) argument
306 static inline PetscErrorCode PetscDeviceContextGetBLASHandle_Internal(PetscDeviceContext dctx, void… in PetscDeviceContextGetBLASHandle_Internal()
315 static inline PetscErrorCode PetscDeviceContextGetSOLVERHandle_Internal(PetscDeviceContext dctx, vo… in PetscDeviceContextGetSOLVERHandle_Internal()
324 static inline PetscErrorCode PetscDeviceContextGetStreamHandle_Internal(PetscDeviceContext dctx, vo… in PetscDeviceContextGetStreamHandle_Internal()
334 static inline PetscErrorCode PetscDeviceContextBeginTimer_Internal(PetscDeviceContext dctx) in PetscDeviceContextBeginTimer_Internal()
343 static inline PetscErrorCode PetscDeviceContextEndTimer_Internal(PetscDeviceContext dctx, PetscLogD… in PetscDeviceContextEndTimer_Internal()
354 static inline PetscErrorCode PetscDeviceContextGetPower_Internal(PetscDeviceContext dctx, PetscLogD… in PetscDeviceContextGetPower_Internal()
364 static inline PetscErrorCode PetscDeviceContextBeginEnergyMeter_Internal(PetscDeviceContext dctx) in PetscDeviceContextBeginEnergyMeter_Internal()
373 static inline PetscErrorCode PetscDeviceContextEndEnergyMeter_Internal(PetscDeviceContext dctx, Pet… in PetscDeviceContextEndEnergyMeter_Internal()
[all …]
/petsc/src/sys/objects/device/tests/
H A Dex9.c5 static PetscErrorCode CheckIdle(PetscDeviceContext dctx, const char operation[]) in CheckIdle()
18 static PetscErrorCode TestQueryIdle(PetscDeviceContext dctx) in TestQueryIdle()
56 PetscDeviceContext dctx = NULL; in main() local
H A Dex6.c6 static PetscErrorCode TestView(PetscDeviceContext dctx) in TestView()
24 PetscDeviceContext dctx, dup; in main() local
H A Dex3.c6 static PetscErrorCode TestPetscDeviceContextDuplicate(PetscDeviceContext dctx) in TestPetscDeviceContextDuplicate()
49 PetscDeviceContext dctx; in main() local
H A Dex7.c20 static PetscErrorCode TestAllocate(PetscDeviceContext dctx, PetscRandom rand, PetscMemType mtype) in TestAllocate()
99 static PetscErrorCode TestAsyncCoherence(PetscDeviceContext dctx, PetscRandom rand) in TestAsyncCoherence()
154 PetscDeviceContext dctx; in main() local
H A Dex4.c47 static PetscErrorCode TestPetscDeviceContextForkJoin(PetscDeviceContext dctx) in TestPetscDeviceContextForkJoin()
74 PetscDeviceContext dctx; in main() local
H A Dpetscdevicetestcommon.h30 static inline PetscErrorCode AssertDeviceContextExists(PetscDeviceContext dctx) in AssertDeviceContextExists()
37 static inline PetscErrorCode AssertDeviceContextDoesNotExist(PetscDeviceContext dctx) in AssertDeviceContextDoesNotExist()
/petsc/src/vec/vec/impls/seq/cupm/
H A Dvecseqcupm.hpp200 inline PetscErrorCode VecCUPMGetArrayAsync_Private(Vec v, PetscScalar **a, PetscDeviceContext dctx)… in VecCUPMGetArrayAsync_Private()
211 …rrorCode VecCUPMRestoreArrayAsync_Private(Vec v, PetscScalar **a, PetscDeviceContext dctx) noexcept in VecCUPMRestoreArrayAsync_Private()
221 inline PetscErrorCode VecCUPMGetArrayAsync(Vec v, PetscScalar **a, PetscDeviceContext dctx = nullpt… in VecCUPMGetArrayAsync()
229 inline PetscErrorCode VecCUPMRestoreArrayAsync(Vec v, PetscScalar **a, PetscDeviceContext dctx = nu… in VecCUPMRestoreArrayAsync()
237 …ErrorCode VecCUPMGetArrayReadAsync(Vec v, const PetscScalar **a, PetscDeviceContext dctx = nullptr… in VecCUPMGetArrayReadAsync()
245 …rCode VecCUPMRestoreArrayReadAsync(Vec v, const PetscScalar **a, PetscDeviceContext dctx = nullptr… in VecCUPMRestoreArrayReadAsync()
253 inline PetscErrorCode VecCUPMGetArrayWriteAsync(Vec v, PetscScalar **a, PetscDeviceContext dctx = n… in VecCUPMGetArrayWriteAsync()
261 inline PetscErrorCode VecCUPMRestoreArrayWriteAsync(Vec v, PetscScalar **a, PetscDeviceContext dctx… in VecCUPMRestoreArrayWriteAsync()
H A Dvecseqcupm_impl.hpp175 inline PetscErrorCode VecSeq_CUPM<T>::CreateSeqCUPM_(Vec v, PetscDeviceContext dctx, PetscScalar *h… in CreateSeqCUPM_()
185 …ointwiseBinary_(BinaryFuncT &&binary, Vec xin, Vec yin, Vec zout, PetscDeviceContext dctx) noexcept in PointwiseBinary_()
217 …Vec, Vec, Vec), BinaryFuncT &&binary, Vec wout, Vec xin, Vec yin, PetscDeviceContext dctx) noexcept in PointwiseBinaryDispatch_()
231 …CUPM<T>::PointwiseUnary_(UnaryFuncT &&unary, Vec xinout, Vec yin, PetscDeviceContext dctx) noexcept in PointwiseUnary_()
292 PetscDeviceContext dctx; in CreateSeqCUPMWithBothArrays() local
307 PetscDeviceContext dctx; in Duplicate() local
323 PetscDeviceContext dctx; in BindToCPU() local
395 PetscDeviceContext dctx; in GetLocalVector() local
446 …de VecSeq_CUPM<T>::AYPXAsync(Vec yin, PetscScalar alpha, Vec xin, PetscDeviceContext dctx) noexcept in AYPXAsync()
503 …de VecSeq_CUPM<T>::AXPYAsync(Vec yin, PetscScalar alpha, Vec xin, PetscDeviceContext dctx) noexcept in AXPYAsync()
[all …]
/petsc/src/mat/impls/dense/seq/cupm/
H A Dmatseqdensecupm.hpp109 PetscDeviceContext dctx; in GetArrayC_() local
120 PetscDeviceContext dctx; in RestoreArrayC_() local
131 PetscDeviceContext dctx; in GetArrayAndMemTypeC_() local
142 PetscDeviceContext dctx; in RestoreArrayAndMemTypeC_() local
185 …PI_Comm comm, PetscInt m, PetscInt n, PetscScalar *data, Mat *A, PetscDeviceContext dctx = nullptr… in MatCreateSeqDenseCUPM()
202 inline PetscErrorCode MatDense_Seq_CUPM<T>::SetPreallocation_(Mat m, PetscDeviceContext dctx, Petsc… in SetPreallocation_()
234 inline PetscErrorCode MatDense_Seq_CUPM<T>::HostToDevice_(Mat m, PetscDeviceContext dctx) noexcept in HostToDevice_()
272 inline PetscErrorCode MatDense_Seq_CUPM<T>::DeviceToHost_(Mat m, PetscDeviceContext dctx) noexcept in DeviceToHost_()
379 PetscDeviceContext dctx; in Factor() local
461 PetscDeviceContext dctx; in Factor() local
[all …]
/petsc/src/ksp/ksp/utils/lmvm/blas_cyclic/cupm/
H A Dblas_cyclic_cupm.cxx32 PetscDeviceContext dctx; in AXPBYCyclic_CUPM_Private() local
57 PetscDeviceContext dctx; in DMVCyclic_CUPM_Private() local
82 PetscDeviceContext dctx; in DSVCyclic_CUPM_Private() local
107 PetscDeviceContext dctx; in TRSVCyclic_CUPM_Private() local
132 PetscDeviceContext dctx; in HEMVCyclic_CUPM_Private() local
157 PetscDeviceContext dctx; in GEMVCyclic_CUPM_Private() local
/petsc/src/ksp/ksp/utils/lmvm/dense/cd_cupm/
H A Dcd_cupm.cxx26 PetscErrorCode UpperTriangular<T>::SolveInPlace(PetscDeviceContext dctx, PetscBool hermitian_transp… in SolveInPlace()
46 PetscErrorCode UpperTriangular<T>::SolveInPlaceCyclic(PetscDeviceContext dctx, PetscBool hermitian_… in SolveInPlaceCyclic()
103 PetscDeviceContext dctx; in MatUpperTriangularSolveInPlace_CUPM() local
130 PetscDeviceContext dctx; in MatUpperTriangularSolveInPlaceCyclic_CUPM() local

123