AMD開源高性能機器智能庫MIOpen

來源 | GitHub

翻譯 | Peter

MIOpen是AMD開源的高性能機器學習庫，目前只支持兩種編程模式：

1. OpenCL

2.HIP

安裝準備

安裝ROCm

基礎軟體棧，其中包括

OpenCL - OpenCL 庫和頭文件

HIP -

HIP 和 HCC庫和頭文件

clang-ocl– 必需

MIOpen依靠miopengemm庫來啟用

ROCm cmake模塊可以從這裡安裝：https://github.com/RadeonOpenCompute/rocm-cmake

對應的安裝說明可在下載的庫中查詢。

cmake配置

第一步：創建目錄：

第二步：配置cmake。MIOpen 的首選後端可以使用-DMIOPEN_BACKEND cmake 變數來設置。

For OpenCL, run:

以上步驟均假設 OpenCL 安裝在某個標準位置。如果沒有，則手動設置這兩個 cmake 變數：

For HIP, run:

cmake 步驟示例：

默認安裝位置設置為 「/opt/rocm」, 可使用 CMAKE_INSTALL_PREFIX設置：

此外， 網路配置的資料庫路徑可由 MIOPEN_DB_PATH 變數設置。默認為資料庫文件的安裝位置。出於開發便捷性， 設置 BUILD_DEV 將改變存儲在源目錄中的資料庫文件的路徑：

使用 ccmake 運行 cmake 後，可以更改配置：

ccmake 程序在 不兼容windows平台。

構建庫

可以使用「Release」配置在build目錄構建庫。

用 「install」命令進行目標安裝：

此時的安裝路徑為已設置的CMAKE_INSTALL_PREFIX。

驅動生成

MIOpen 提供了一個應用驅動程序，可用於單獨執行任何一個特定的層，並測量和驗證庫的性能。

可以使用 MIOpenDriver 目標構建驅動程序：

有關如何運行驅動程序的文檔在這裡：https://github.com/ROCmSoftwarePlatform/MIOpen/blob/master/driver/README.md

運行測試

可以使用 target check 運行測試：

通過執行以下操作， 可以生成和運行單個測試：

生成文檔

HTML和PDF文檔生成命令：

這將在./MIOpen/doc/html 文件夾建立一個本地可搜索網站， 並在./MIOpen/doc/pdf 文件夾創建一個 PDF 文檔。

文檔是使用Doxygen生成的， 應單獨安裝。

HTML和PDF通過Sphinx和Breathe以及ReadTheDocs theme生成。

可以在MIOpen / doc文件夾中填寫Sphinx，Breathe和ReadTheDocs theme的要求：

由於設置不同，有可能需要執行sudo才能安裝pip 。

代碼格式化

代碼格式化使用 clang-format。格式化一個文件，使用：

此外，可以安裝githook來格式化per-commit代碼：

點擊展開全文

喜歡這篇文章嗎？立刻分享出去讓更多人知道吧！