CudaMgr_Namespace::DeviceMemoryAllocationMap Class Reference

#include <DeviceMemoryAllocationMap.h>

Classes
struct	Allocation

Public Types
using	DevicePtr = uint64_t
using	Map = std::map< DevicePtr, Allocation >
using	MapChangedCBID = uint32_t
using	MapChangedCB = std::function< void(const heavyai::UUID, const bool)>

Public Member Functions
	DeviceMemoryAllocationMap ()
	~DeviceMemoryAllocationMap ()=default
const Map &	getMap () const
const bool	mapEmpty () const
void	addAllocation (const DevicePtr device_ptr, const uint64_t size, const uint64_t handle, const heavyai::UUID uuid, const int device_num, const bool is_slab)
Allocation	removeAllocation (const DevicePtr device_ptr)
std::pair< DevicePtr, Allocation >	getAllocation (const DevicePtr device_ptr)
const MapChangedCBID	registerMapChangedCB (MapChangedCB cb)
void	unregisterMapChangedCB (const MapChangedCBID cbid)
void	notifyMapChanged (const heavyai::UUID device_uuid, const bool is_slab) const

Private Attributes
Map	map_
MapChangedCBID	last_map_changed_cbid_
std::unordered_map < MapChangedCBID, MapChangedCB >	map_changed_cbs_

Detailed Description

Definition at line 28 of file DeviceMemoryAllocationMap.h.

Member Typedef Documentation

using CudaMgr_Namespace::DeviceMemoryAllocationMap::DevicePtr = uint64_t

Definition at line 30 of file DeviceMemoryAllocationMap.h.

using CudaMgr_Namespace::DeviceMemoryAllocationMap::Map = std::map<DevicePtr, Allocation>

Definition at line 38 of file DeviceMemoryAllocationMap.h.

using CudaMgr_Namespace::DeviceMemoryAllocationMap::MapChangedCB = std::function<void(const heavyai::UUID, const bool)>

Definition at line 40 of file DeviceMemoryAllocationMap.h.

using CudaMgr_Namespace::DeviceMemoryAllocationMap::MapChangedCBID = uint32_t

Definition at line 39 of file DeviceMemoryAllocationMap.h.

Constructor & Destructor Documentation

CudaMgr_Namespace::DeviceMemoryAllocationMap::DeviceMemoryAllocationMap

(

)

Definition at line 24 of file DeviceMemoryAllocationMap.cpp.

: last_map_changed_cbid_{0u} {}

CudaMgr_Namespace::DeviceMemoryAllocationMap::~DeviceMemoryAllocationMap ( )

default

Member Function Documentation

void CudaMgr_Namespace::DeviceMemoryAllocationMap::addAllocation	(	const DevicePtr	device_ptr,
		const uint64_t	size,
		const uint64_t	handle,
		const heavyai::UUID	uuid,
		const int	device_num,
		const bool	is_slab
	)

Definition at line 34 of file DeviceMemoryAllocationMap.cpp.

References CHECK, and map_.

                                                                  {
  Allocation allocation{size, handle, uuid, device_num, is_slab};
  CHECK(map_.try_emplace(device_ptr, std::move(allocation)).second);
}

std::pair< DeviceMemoryAllocationMap::DevicePtr, DeviceMemoryAllocationMap::Allocation > CudaMgr_Namespace::DeviceMemoryAllocationMap::getAllocation

(

const DevicePtr

device_ptr

)

Definition at line 58 of file DeviceMemoryAllocationMap.cpp.

References CHECK, and map_.

                                                                   {
  // find the map entry above this address
  auto itr = map_.upper_bound(device_ptr);
  CHECK(itr != map_.begin());
  // return the previous entry
  --itr;
  return std::make_pair(itr->first, itr->second);
}

const DeviceMemoryAllocationMap::Map & CudaMgr_Namespace::DeviceMemoryAllocationMap::getMap

(

)

const

Definition at line 26 of file DeviceMemoryAllocationMap.cpp.

References map_.

                                                                            {
  return map_;
}

const bool CudaMgr_Namespace::DeviceMemoryAllocationMap::mapEmpty

(

)

const

Definition at line 30 of file DeviceMemoryAllocationMap.cpp.

References map_.

                                                     {
  return map_.empty();
}

void CudaMgr_Namespace::DeviceMemoryAllocationMap::notifyMapChanged	(	const heavyai::UUID	device_uuid,
		const bool	is_slab
	)		const

Definition at line 80 of file DeviceMemoryAllocationMap.cpp.

References map_changed_cbs_.

                                                                           {
  for (auto const& cb : map_changed_cbs_) {
    cb.second(device_uuid, is_slab);
  }
}

const DeviceMemoryAllocationMap::MapChangedCBID CudaMgr_Namespace::DeviceMemoryAllocationMap::registerMapChangedCB

(

MapChangedCB

cb

)

Definition at line 68 of file DeviceMemoryAllocationMap.cpp.

References CHECK, last_map_changed_cbid_, and map_changed_cbs_.

                                                               {
  auto const cbid = ++last_map_changed_cbid_;
  CHECK(map_changed_cbs_.emplace(cbid, cb).second) << "Repeat registration";
  return cbid;
}

DeviceMemoryAllocationMap::Allocation CudaMgr_Namespace::DeviceMemoryAllocationMap::removeAllocation

(

const DevicePtr

device_ptr

)

Definition at line 44 of file DeviceMemoryAllocationMap.cpp.

References CHECK, and map_.

                                {
  // find the exact match
  auto const itr = map_.find(device_ptr);
  CHECK(itr != map_.end());
  // copy out the allocation
  Allocation allocation = itr->second;
  // remove from map
  map_.erase(itr);
  // return the allocation that was
  return allocation;
}

void CudaMgr_Namespace::DeviceMemoryAllocationMap::unregisterMapChangedCB

(

const MapChangedCBID

cbid

)

Definition at line 74 of file DeviceMemoryAllocationMap.cpp.

References CHECK, and map_changed_cbs_.

                                                                                {
  auto itr = map_changed_cbs_.find(cbid);
  CHECK(itr != map_changed_cbs_.end()) << "Failed to unregister";
  map_changed_cbs_.erase(itr);
}

Member Data Documentation

MapChangedCBID CudaMgr_Namespace::DeviceMemoryAllocationMap::last_map_changed_cbid_

private

Definition at line 63 of file DeviceMemoryAllocationMap.h.

Referenced by registerMapChangedCB().

Map CudaMgr_Namespace::DeviceMemoryAllocationMap::map_

private

Definition at line 62 of file DeviceMemoryAllocationMap.h.

Referenced by addAllocation(), getAllocation(), getMap(), mapEmpty(), and removeAllocation().

std::unordered_map<MapChangedCBID, MapChangedCB> CudaMgr_Namespace::DeviceMemoryAllocationMap::map_changed_cbs_

private

Definition at line 64 of file DeviceMemoryAllocationMap.h.

Referenced by notifyMapChanged(), registerMapChangedCB(), and unregisterMapChangedCB().

---

The documentation for this class was generated from the following files:

• /home/jenkins-slave/workspace/core-os-doxygen/CudaMgr/DeviceMemoryAllocationMap.h
• /home/jenkins-slave/workspace/core-os-doxygen/CudaMgr/DeviceMemoryAllocationMap.cpp