Lines Matching refs:DeviceContext
27 class PETSC_SINGLE_LIBRARY_VISIBILITY_INTERNAL DeviceContext : SolverInterface<T> { class
262 inline PetscErrorCode DeviceContext<T>::initialize(PetscDevice device) noexcept in initialize()
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()
346 inline PetscErrorCode DeviceContext<T>::waitForContext(PetscDeviceContext dctxa, PetscDeviceContext… in waitForContext()
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()
392 inline PetscErrorCode DeviceContext<T>::beginTimer(PetscDeviceContext dctx) noexcept in beginTimer()
412 inline PetscErrorCode DeviceContext<T>::endTimer(PetscDeviceContext dctx, PetscLogDouble *elapsed) … in endTimer()
433 inline PetscErrorCode DeviceContext<T>::getPower(PetscDeviceContext dctx, PetscLogDouble *power) no… in getPower()
451 inline PetscErrorCode DeviceContext<T>::beginEnergyMeter(PetscDeviceContext dctx) noexcept in beginEnergyMeter()
467 inline PetscErrorCode DeviceContext<T>::endEnergyMeter(PetscDeviceContext dctx, PetscLogDouble *ene… in endEnergyMeter()
485 inline PetscErrorCode DeviceContext<T>::memAlloc(PetscDeviceContext dctx, PetscBool clear, PetscMem… in memAlloc()
502 inline PetscErrorCode DeviceContext<T>::memFree(PetscDeviceContext dctx, PetscMemType mtype, void *… in memFree()
532 inline PetscErrorCode DeviceContext<T>::memCopy(PetscDeviceContext dctx, void *PETSC_RESTRICT dest,… in memCopy()
558 inline PetscErrorCode DeviceContext<T>::memSet(PetscDeviceContext dctx, PetscMemType mtype, void *p… in memSet()
568 inline PetscErrorCode DeviceContext<T>::createEvent(PetscDeviceContext, PetscEvent event) noexcept in createEvent()
582 inline PetscErrorCode DeviceContext<T>::recordEvent(PetscDeviceContext dctx, PetscEvent event) noex… in recordEvent()
590 inline PetscErrorCode DeviceContext<T>::waitForEvent(PetscDeviceContext dctx, PetscEvent event) noe… in waitForEvent()
599 bool DeviceContext<T>::initialized_ = false;
602 std::array<typename DeviceContext<T>::cupmBlasHandle_t, PETSC_DEVICE_MAX_DEVICES> DeviceContext<T>:…
605 std::array<typename DeviceContext<T>::cupmSolverHandle_t, PETSC_DEVICE_MAX_DEVICES> DeviceContext<T…
608 constexpr _DeviceContextOps DeviceContext<T>::ops;
613 using CUPMContextCuda = impl::DeviceContext<DeviceType::CUDA>;
614 using CUPMContextHip = impl::DeviceContext<DeviceType::HIP>;
617 #define PetscDeviceContext_(IMPLS) ::Petsc::device::cupm::impl::DeviceContext<::Petsc::device::cupm…