mxnet 源码阅读 2

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 都应该视为错误,而这里面可能逻辑稍微有点问题

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

posted on 2017-08-08 12:42  reedlau  阅读(595)  评论(0编辑  收藏  举报

导航