1. 程式人生 > >mxnet 源碼閱讀 1

mxnet 源碼閱讀 1

ces 區別 值類型 tor win 負責 字節對齊 平臺 align

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