嗨,我想把我的一些CUDA内核函数分离到一个单独的文件中,这样我就可以重用它们了。
假设我有两份文件:
A.cu包含可重用的CUDA内核。B.cu包含一些内核和主机函数,我想在其中调用A.cu文件中的一些内核。我该怎么做呢?
发布于 2014-11-17 18:09:46
您可以做的是将内核原型放在一个.cuh文件中,然后将其包含在第二个文件中。Here是组织你的数据自动化系统代码的一种方式。
发布于 2014-11-17 18:06:35
对于您所描述的情况,您可以以几乎与在C/C++中进行此操作的方式相同的方式来完成此操作。下面是一个充分发挥作用的例子:
$ cat B.cu
#include "myheader.h"
__global__ void kernel1(){
printf("Hello 1\n");
}
int main(){
kernel1<<<1,1>>>();
cudaDeviceSynchronize();
kernel2<<<1,1>>>();
cudaDeviceSynchronize();
return 0;
}
$ cat A.cu
#include "myheader.h"
__global__ void kernel2(){
printf("Hello 2\n");
}
$ cat myheader.h
#include <stdio.h>
__global__ void kernel2();
$ nvcc -arch=sm_20 -o test A.cu B.cu
$ cuda-memcheck ./test
========= CUDA-MEMCHECK
Hello 1
Hello 2
========= ERROR SUMMARY: 0 errors
$https://stackoverflow.com/questions/26978470
复制相似问题