ROCT-Thunk-Interface
是一个用于在ROCm软件堆栈中提供设备无关性的层。它是ROCm的一部分,允许不同的硬件平台(如AMD GPU和Intel CPU)使用相同的API进行计算。
要安装ROCT-Thunk-Interface
,首先需要创建一个新的目录,并进入该目录:
mkdir rocm-build
cd rocm-build
然后,克隆ROCm的GitHub存储库:
git clone https://github.com/RadeonOpenCompute/ROCm.git
cd ROCm
手动添加ROCT-Thunk-Interface子模块:
git submodule add https://github.com/RadeonOpenCompute/ROCT-Thunk-Interface.git packages/ROCT-Thunk-Interface
现在可以开始构建和安装ROCT-Thunk-Interface,进入ROCT-Thunk-Interface目录,编译出来libhsakmt.a:
mkdir build
cd build
cmake .. \
-DCMAKE_BUILD_TYPE=Release \
-DCMAKE_INSTALL_PREFIX=/opt/rocm \
-DROCTHUNKINSTALLPREFIX=/opt/rocm/thunk
make -j 12
sudo make install
这将在/opt/rocm/thunk
目录下安装ROCT-Thunk-Interface
。
编译KFDTEST:
进入/home/love/Workspace/zhai/rocm-build/ROCm/packages/ROCT-Thunk-Interface/tests/kfdtest:
mkdir build
cd build
cmake ../
make -j 12
可以看到编译出来了kfdtest程序。
运行测试:
$ sudo ./run_kfdtest.sh all
参考文章
GitCode - 开发者的代码家园
基于AMDGPU-ROCm的深度学习环境搭建_amd显卡深度学习-CSDN博客