首页 > 代码库 > mxnet 源码阅读 1
mxnet 源码阅读 1
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
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。