在.c文件中调用cuda函数

问题描述:假设在Ubuntu的一个用户目录下有2个文件,main.c, VectorAdd.cu,其中 VectorAdd.cu有vectorAdd函数,main.c提供程序的入口main函数。现在为了在main.c中实现两个向量相加的操作,就需要调用 VectorAdd.cu中的vectorAdd函数

首先列出两个文件中的内容


//VectorAdd.cu
#include <cutil.h>
extern "C" void VectAdd(int *a, int *b, int *c, int length);

__global__ void Add(int *d_a, int *d_b, int *d_c, int length)
{
	int id = threadIdx.x;
	if(id < length)
		d_c[id] = d_a[id] + d_b[id];
}

void VectAdd(int *a, int *b, int *c, int length)
{	
	unsigned int size = sizeof(int) * length; 
	int *d_a;
	cudaMalloc((void**)&d_a,size);
	int *d_b;
	cudaMalloc((void**)&d_b,size);
	int *d_c;
	cudaMalloc((void**)&d_c,size);

	cudaMemcpy(d_a, a, size, cudaMemcpyHostToDevice);
	cudaMemcpy(d_b, b, size, cudaMemcpyHostToDevice);
	
	Add<<<1, length>>>(d_a, d_b, d_c, length);

	cudaMemcpy(c, d_c, size, cudaMemcpyDeviceToHost);

	cudaFree(d_a);
	cu
  • 3
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值