使用 Meson 构建 CUDA 内核

Question

我正在尝试构建一个使用 Cuda 内核的应用程序。为此，我正在使用meson build system，但没有成功。基本上我想做的是以下几点：

//main.cpp

extern void kernel_function();

int main(int argc, char *argv[]){

  // some logic here...

  kernel_function();
  return 0;
}

//kernel.cu

__global__ void kernel(){
  // some code here...
}

void kernel_function(){

  dim3 threads( 2, 1 );
  dim3 blocks( 1, 1 );

  kernel<<< blocks, threads >>>();
}

我可以用以下命令编译上面的代码：

g++ -c main.cpp
nvcc -c kernel.cu
nvcc -o main main.o kernel.o

如何复制我在终端上使用 Meson 执行的编译过程？

Answer 1

在最简单的情况下，它可能是这样的：

project('cuda dependency', 'cpp', 'cuda')
executable('main', 'main.cpp', 'kernel.cu')

然后运行

$ CXX=g++ meson build
$ ninja -C build

（基于example）。

但你可能需要添加依赖项，例如

dep = dependency('cuda', version : '>=10', modules : ['cublas'])
executable(..., dependencies: dep)

或借助 cuda 模块设置一些特殊的编译器标志，检查介子上的this reference。