linux - 使用 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在介子上。