CUDA的函数指针

[adagio_chen]

函数指针是C/C++中很有意思的一个特性。以下是一个简单的例子：

typedef int (*FP)(int);

int Fun(int n)
{
	return n*2;
}

void TestFP(int n, FP fun)
{
	printf("%d\n", fun(n));
}

FP fp = Fun;

int main()
{
	TestFP(5, fp);
	return 0;
}

目前的CUDA版本也支持函数指针。于是有人可能会简单的把上述代码改成GPU版本：

typedef int (*FP)(int);

__device__
int Fun(int n)
{
	return n*2;
}

__global__
void TestFP(int n, FP fun)
{
	printf("%d\n", fun(n));
}

__device__ FP fp = Fun;

int main()
{
	TestFP<<<1,1>>>(5, fp);
	return 0;
}

然而，这种做法是错误的，它不能得到任何结果。错误的原因在于：无法给__global__函数传递一个__device__函数指针。而正确的做法是给它传递一个host函数指针：

typedef int (*FP)(int);

__device__
int Fun(int n)
{
	return n*2;
}

__global__
void TestFP(int n, FP fun)
{
	printf("%d\n", fun(n));
}

__device__ FP fp_d = Fun;
FP fp_h;

int main()
{
	cudaMemcpyFromSymbol(&fp_h, fp_d, sizeof(FP));
	TestFP<<<1,1>>>(5, fp_h);
	return 0;
}

[qq_21595381]

这是。。有点看不懂。

[gyouan]

很受用,有学习到了新的知识

[jzelen]

这个属实没太看懂。

[太好欺负]

这个是函数回调 吗

[qq_24800305]

难道是函数回调？

[zhayuxing]

真是牛人啊啊

[prettyboyllz]

感谢版主分享

[GW786228836]

[foreverccc]

感谢版主分享

[xusir98]

[ygawow]

我是来学习的

[g13863832110]

[qq_24811003]

很好的资源~非常感谢！

[qq_24833445]

高手 哇

[qiuserudan]

很有用的东西，学习了！

[yoyojoker123]

看不懂看不懂

[woshiaoo]

感谢版主分享

[sunny061009]

学习了,很好的代码

[2220728]

又是MSB3721的错误，楼主什么情况，有什么必要的头文件包含吗

[aerospacex]

编译时报错：1个无法解析的外部命令，这是怎么回事呢