26 template <DeviceType T>
27 class PETSC_SINGLE_LIBRARY_VISIBILITY_INTERNAL DeviceContext : SolverInterface<T> {
29   PETSC_CUPMSOLVER_INHERIT_INTERFACE_TYPEDEFS_USING(T);
41   using stream_type = CUPMStream<T>;
42   using event_type  = CUPMEvent<T>;
83 …C_NODISCARD static constexpr CUPMEvent<T> *event_cast_(PetscEvent event) noexcept { return static_…  in event_cast_()
85 …PETSC_NODISCARD static PetscLogEvent CUPMBLAS_HANDLE_CREATE() noexcept { return T == DeviceType::C…  in CUPMBLAS_HANDLE_CREATE()
87 …PETSC_NODISCARD static PetscLogEvent CUPMSOLVER_HANDLE_CREATE() noexcept { return T == DeviceType:…  in CUPMSOLVER_HANDLE_CREATE()
261 template <DeviceType T>
262 inline PetscErrorCode DeviceContext<T>::initialize(PetscDevice device) noexcept  in initialize()
279 template <DeviceType T>
280 inline PetscErrorCode DeviceContext<T>::destroy(PetscDeviceContext dctx) noexcept  in destroy()
285     if (dci->event) PetscCall(cupm_fast_event_pool<T>().deallocate(&dci->event));  in destroy()
294 template <DeviceType T>
295 inline PetscErrorCode DeviceContext<T>::changeStreamType(PetscDeviceContext dctx, PETSC_UNUSED Pets…  in changeStreamType()
307 template <DeviceType T>
308 inline PetscErrorCode DeviceContext<T>::setUp(PetscDeviceContext dctx) noexcept  in setUp()
316   if (!event) PetscCall(cupm_fast_event_pool<T>().allocate(&event));  in setUp()
323 template <DeviceType T>
324 inline PetscErrorCode DeviceContext<T>::query(PetscDeviceContext dctx, PetscBool *idle) noexcept  in query()
345 template <DeviceType T>
346 inline PetscErrorCode DeviceContext<T>::waitForContext(PetscDeviceContext dctxa, PetscDeviceContext…  in waitForContext()
358 template <DeviceType T>
359 inline PetscErrorCode DeviceContext<T>::synchronize(PetscDeviceContext dctx) noexcept  in synchronize()
369 template <DeviceType T>
371 inline PetscErrorCode DeviceContext<T>::getHandle(PetscDeviceContext dctx, void *handle) noexcept  in getHandle()
379 template <DeviceType T>
381 inline PetscErrorCode DeviceContext<T>::getHandlePtr(PetscDeviceContext dctx, void **handle) noexce…  in getHandlePtr()
391 template <DeviceType T>
392 inline PetscErrorCode DeviceContext<T>::beginTimer(PetscDeviceContext dctx) noexcept  in beginTimer()
411 template <DeviceType T>
412 inline PetscErrorCode DeviceContext<T>::endTimer(PetscDeviceContext dctx, PetscLogDouble *elapsed) …  in endTimer()
432 template <DeviceType T>
433 inline PetscErrorCode DeviceContext<T>::getPower(PetscDeviceContext dctx, PetscLogDouble *power) no…  in getPower()
450 template <DeviceType T>
451 inline PetscErrorCode DeviceContext<T>::beginEnergyMeter(PetscDeviceContext dctx) noexcept  in beginEnergyMeter()
466 template <DeviceType T>
467 inline PetscErrorCode DeviceContext<T>::endEnergyMeter(PetscDeviceContext dctx, PetscLogDouble *ene…  in endEnergyMeter()
484 template <DeviceType T>
485 inline PetscErrorCode DeviceContext<T>::memAlloc(PetscDeviceContext dctx, PetscBool clear, PetscMem…  in memAlloc()
493 …PetscCall(default_pool_<HostAllocator<T>>().allocate(n, reinterpret_cast<char **>(dest), &stream, …  in memAlloc()
495 …PetscCall(default_pool_<DeviceAllocator<T>>().allocate(n, reinterpret_cast<char **>(dest), &stream…  in memAlloc()
501 template <DeviceType T>
502 inline PetscErrorCode DeviceContext<T>::memFree(PetscDeviceContext dctx, PetscMemType mtype, void *…  in memFree()
511 …  PetscCall(default_pool_<HostAllocator<T>>().deallocate(reinterpret_cast<char **>(ptr), &stream));  in memFree()
524 …PetscCall(default_pool_<DeviceAllocator<T>>().deallocate(reinterpret_cast<char **>(ptr), &stream));  in memFree()
531 template <DeviceType T>
532 inline PetscErrorCode DeviceContext<T>::memCopy(PetscDeviceContext dctx, void *PETSC_RESTRICT dest,…  in memCopy()
557 template <DeviceType T>
558 inline PetscErrorCode DeviceContext<T>::memSet(PetscDeviceContext dctx, PetscMemType mtype, void *p…  in memSet()
567 template <DeviceType T>
568 inline PetscErrorCode DeviceContext<T>::createEvent(PetscDeviceContext, PetscEvent event) noexcept  in createEvent()
581 template <DeviceType T>
582 inline PetscErrorCode DeviceContext<T>::recordEvent(PetscDeviceContext dctx, PetscEvent event) noex…  in recordEvent()
589 template <DeviceType T>
590 inline PetscErrorCode DeviceContext<T>::waitForEvent(PetscDeviceContext dctx, PetscEvent event) noe…  in waitForEvent()
598 template <DeviceType T>
599 bool DeviceContext<T>::initialized_ = false;
601 template <DeviceType T>
602 std::array<typename DeviceContext<T>::cupmBlasHandle_t, PETSC_DEVICE_MAX_DEVICES> DeviceContext<T>:…
604 template <DeviceType T>
605 std::array<typename DeviceContext<T>::cupmSolverHandle_t, PETSC_DEVICE_MAX_DEVICES> DeviceContext<T…
607 template <DeviceType T>
608 constexpr _DeviceContextOps DeviceContext<T>::ops;