说明
由于实验室项目原因,需要跑一个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库的安装路径
至此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
下面。