device/interface/petscdevice_interface_internal.hpp

5 #include <petsc/private/cpp/utility.hpp> // std::pair
6 #include <petsc/private/cpp/memory.hpp>  // std::weak_ptr, std::shared_ptr
9 #include <algorithm> // std::lower_bound
14 // unordered_set for clang and hand-rolled flat set for GCC...
45   using weak_ptr_type = std::weak_ptr<_p_PetscDeviceContext>;
51     using std::swap;  in swap()
71 …t _n_WeakContext(const std::shared_ptr<_p_PetscDeviceContext> &ptr) noexcept : weak_dctx_{ptr}, st…  in _n_WeakContext()
80   using upstream_type = std::unordered_map<PetscObjectId, _n_WeakContext>;
82   using marked_type = std::unordered_set<PetscObjectId>;
84   using marked_type = std::vector<PetscObjectId>;
86   using shared_ptr_type = std::shared_ptr<_p_PetscDeviceContext>;
104 …PETSC_NODISCARD std::pair<bool, typename marked_type::iterator> get_marked_(PetscObjectId id) noex…  in get_marked_()
106     auto end = this->marked_objects().end();  in get_marked_()
107     auto it  = std::lower_bound(this->marked_objects().begin(), end, id);  in get_marked_()
132   PetscCallCXX(this->upstream().clear());  in clear()
133   PetscCallCXX(this->marked_objects().clear());  in clear()
139   return _n_WeakContext{this->self()};  in weak_snapshot()
160   return const_cast<CxxData *>(this)->get_marked_(id).first;  in has_marked()
171   return static_cast<CxxData *>(PetscObjectCast(dctx)->cpp);  in CxxDataCast()
179 …ns_Internal(PetscOptionItems PetscOptionsObject, std::pair<PetscDeviceType, PetscBool> &deviceType…  in PetscDeviceContextQueryOptions_Internal()
186 …PetscCall(PetscOptionsEList("-device_context_device_type", "Underlying PetscDevice", "PetscDeviceC…  in PetscDeviceContextQueryOptions_Internal()
187 …PetscCall(PetscOptionsEList("-device_context_stream_type", "PetscDeviceContext PetscStreamType", "…  in PetscDeviceContextQueryOptions_Internal()