mxnet 源码阅读 2

标签：检测   err   空间   success   阅读   src   ror   日志   申请   

mxnet/src/storage/gpu_device_storage.h

同cpu_device_storage.h 类似通过两级命名空间 mxnet 和 storage 定义了类 GPUDeviceStorage ，内部有两个静态成员函数

Alloc 和 Free 其实现通过调用CUDA API cudaMalloc 和 cudaFree 实现 ，同时对于内存申请和释放出错的情况写出了日志信息，CPU部分是没有日志的。

个人感觉include的头文件有点大了，可以更细化的只include logging.h 而已。

同时对于cuda返回错误检测多了个 cudaErrorCudartUnloading 不太接受，官方文档给出的也就两个值 对于！cudaSuccess 都应该视为错误，而这里面可能逻辑稍微有点问题

即使是硬件驱动错误也应该是错误的，但是这里处理的是没有出错。

mxnet 源码阅读 2

标签：检测   err   空间   success   阅读   src   ror   日志   申请   

原文地址：http://www.cnblogs.com/reedlau/p/7306240.html