版主及大牛们,你们好!小菜想请教一个问题:
如果我调用两个内核函数,第二内核函数要用到第一个内核函数处理的结果,那么我能否直接将指针作为第二个内函数的参数进行计算吗?还是说我要先将第一个内核函数的结果复制到CPU,然后在开辟一个新的显存空间,把数据传过去?在线等!
谢谢!
LZ你好,可以直接将指针作为第二个内核函数的参数,不需要将结果copy回CPU内存。举例如下:
kernel1(float* in,float * out){}
kernel2(float* in,float* out){}
main(){
…
kernel1<<<>>>(dev_pa,dev_pb);
kernel2<<<>>>(dev_pb,dev_pc);
…
}
好的!谢谢!我做个实验试试!有问题再请教