mxnet 源碼閱讀 1

來源:互聯網
上載者:User

標籤:靜態函數   源碼   名字空間   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

相關文章

聯繫我們

該頁面正文內容均來源於網絡整理,並不代表阿里雲官方的觀點,該頁面所提到的產品和服務也與阿里云無關,如果該頁面內容對您造成了困擾,歡迎寫郵件給我們,收到郵件我們將在5個工作日內處理。

如果您發現本社區中有涉嫌抄襲的內容,歡迎發送郵件至: info-contact@alibabacloud.com 進行舉報並提供相關證據,工作人員會在 5 個工作天內聯絡您,一經查實,本站將立刻刪除涉嫌侵權內容。

A Free Trial That Lets You Build Big!

Start building with 50+ products and up to 12 months usage for Elastic Compute Service

  • Sales Support

    1 on 1 presale consultation

  • After-Sales Support

    24/7 Technical Support 6 Free Tickets per Quarter Faster Response

  • Alibaba Cloud offers highly flexible support services tailored to meet your exact needs.