大家好，又见面了，我是全栈君。
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 都应该视为错误，而这里面可能逻辑稍微有点问题
即使是硬件驱动错误也应该是错误的，但是这里处理的是没有出错。
转载于:https://www.cnblogs.com/reedlau/p/7306240.html
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 举报，一经查实，本站将立刻删除。
发布者：全栈程序员-用户IM，转载请注明出处：https://javaforall.cn/108260.html原文链接：https://javaforall.cn
【正版授权，激活自己账号】： Jetbrains全家桶Ide使用，1年售后保障，每天仅需1毛
【官方授权 正版激活】： 官方授权 正版激活 支持Jetbrains家族下所有IDE 使用个人JB账号...
