mxnet 源码阅读 2
2021-06-08 22:02
阅读：470
标签：检测 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
下一篇：WebSocket简单介绍
评论
亲，登录后才可以留言！