码迷,mamicode.com
首页 > Web开发 > 详细

mxnet 源码阅读 1

时间:2017-08-08 13:58:23      阅读:178      评论:0      收藏:0      [点我收藏+]

标签:静态函数   源码   名字空间   udev   alignment   lin   定义   net   dev   

mxnet/src/storage/cpu_device_storage.h

通过两个名字空间 mxnet 和 storage 定义了CPUDeviceStorage类,然后在其中定义了两个静态函数 Alloc 和 Free 负责申请CPU内存,内存是按照 alignment_=16 对齐的

用 宏 _MSC_VER 区分了Windows 平台和 Linux/Unix 平台,Windows 用 _aligned_malloc 申请特定字节对齐的内存 ; Linux 等用Posix 提供的  posix_memalign 申请,

这两个函数的返回值类型不同,要区别对待。

 

mxnet 源码阅读 1

标签:静态函数   源码   名字空间   udev   alignment   lin   定义   net   dev   

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

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!