mxnet代码剖析之--Storage篇

/// Storage manager across multiple devices

struct Handle

void* dptr; /// Pointer to the data.

size_t size; /// Size of the storage.

Context ctx; /// Context information about device and ID.

/// Base storage manager interface

class StorageManager

Alloc()

Free()

class NaiveStorageManager, PooledStorageManager

class CPUDeviceStorage, GPUDeviceStorage, PinnedMemoryStorage

class LazyAllocArray: Get element of corresponding index, if it is not created create by creator

class MXNET_API Storage

1 mxnet存储管理器

2 存储由内存地址，大小，设备属性构成，分别对应Handle: dptr, size, ctx

3 存储管理器主要负责内存申请，内存销毁操作

4 实现内存复用，节省内存new/delete引起的开销，主要由PooledStorageManager类完成

5 存储设备包括GPU, CPU, PinnedCPU, 具体实现分别由GPUDeviceStorage, CPUDeviceStorage与PinnedMemoryStorage实现

6 优点：封装内存管理类，尤其是pooled管理策略！结合其LazyAllocArray功能，可以实现内存复用，减少内存new/delete 操作