大家好，又见面了，我是全栈君。
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
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请联系我们举报，一经查实，本站将立刻删除。
发布者：全栈程序员-站长，转载请注明出处：https://javaforall.net/108260.html原文链接：https://javaforall.net
