CK-Composable Kernel 下載安裝
AMD 可組合核心庫：只需幾行程式碼即可為 AI 應用提供高效的融合核心
CK 庫是為 AMD-GPU 所開發的，下載和使用都需要在 AMD-Rocm 環境/平臺，如果你的 GPU 並非 AMD 的，是無法下載使用 CK 的。
關於CK庫
下載
從 GitHub 儲存庫下載 CK 原始碼並開始構建。 將會得到一個名為composable_kernel的資料夾。
git clone https://github.com/ROCm/composable_kernel.git
cd composable_kernel
安裝 (Python + Cmake)
安裝Python依賴
cd docs
pip3 install -r sphinx/requirements.txt
python3 -m sphinx -T -E -b html -d _build/doctrees -D language=en . _build/html
檢視GPU架構
CK支援的GPU架構有 gfx908;gfx90a;gfx940;gfx941;gfx942;gfx1030;gfx1100;gfx1101;gfx1102
需要檢視自己本地(或你的伺服器)使用的架構,以下兩行程式碼都可以幫助你明確你所使用的架構。
rocminfo
rocm-smi
我的GPU架構是 gfx942 。因此在下面的cmake配置中，我的 GPU_TARGETS 項為 gfx942 。
使用Cmake安裝
建立構建目錄並進入。
mkdir build
cd build
使用以下命令配置cmake, 注意路徑，並把 GPU_TARGETS 項更改為你的GPU架構。
cmake \
-D CMAKE_PREFIX_PATH=/opt/rocm \
-D CMAKE_CXX_COMPILER=/opt/rocm/bin/hipcc \
-D CMAKE_BUILD_TYPE=Release \
-D GPU_TARGETS="gfx942" \
..
構建CK庫 並安裝CK
make -j
make -j install
示例測試
生成示例和測試
make -j examples tests
生成並執行所有示例和測試
make -j check
- CK庫使用介紹
- AMD ROCm 開放軟體平臺