mxnet 源碼閱讀 2

來源:互聯網
上載者:User

標籤:檢測   err   空間   success   閱讀   src   ror   日誌   申請   

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 都應該視為錯誤,而這裡面可能邏輯稍微有點問題

即使是硬體驅動錯誤也應該是錯誤的,但是這裡處理的是沒有出錯。

mxnet 源碼閱讀 2

相關文章

聯繫我們

該頁面正文內容均來源於網絡整理,並不代表阿里雲官方的觀點,該頁面所提到的產品和服務也與阿里云無關,如果該頁面內容對您造成了困擾,歡迎寫郵件給我們,收到郵件我們將在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.