526互联

Linux服务器安装Kokkos-core 和 Kokkos-kernel

发布时间 2023-05-26 12:08:31作者: 人生逆旅，我亦行人

说明

由于实验室项目原因，需要跑一个Gmres算法，之前弄过kokkos，就想在kokkos-kernels里跑现有的GMRES算法库
在此记录自己的安装的过程，以及自己踩过的一些坑。

1. 准备工作

从Kokkos官网下载Kokkos 以及 Kokkos-kernels：
https://github.com/kokkos/kokkos.git -- Kokkos-core
https://github.com/kokkos/kokkos-kernels.git -- Kokkos-kernels
注意： 如果是压缩包下载到服务器上的话，编译Kokkos-core的时候需要先将Kokkos-master/bin/里面的四个脚本加上可执行权限，不然编译的时候会报错。

2. 安装Kokkos-core

然后在kokkos-master/目录下

mkdir build && cd build
cmake ..  // 若要指定CUDA后端则加上编译选项 -DKokkos_ENABLE_CUDA=ON -DKokkos_ARCH_PASCAL61=ON（可在https://kokkos.github.io/kokkos-core-wiki/keywords.html找到相应的编译选项 ）
make -j
make install // 可能要加上sudo权限

make install 的时候可以看下路径，后面编译kokkos-kernel的时候需要指定Kokkos库的安装路径
image
至此Kokkos-core就已经安装好了

3. 安装Kokkos-kernel

在目录kokkos-kernels-master目录下

mkdir build && cd build
cmake .. -DCMAKE_BUILD_TYPE=Release -DKokkosKernels_ENABLE_TESTS=ON -DKokkosKernels_ENABLE_EXAMPLES=ON -DKokkos_ROOT=/usr/local -DKokkos_ENABLE_CUDA=ON // 此处我指定后端为cuda并且指定相应的架构
make -j

然后在 build目录下，example中就有相关test代码啦，我此次测试的是gmres就在build/example/gmres下面。
image