cuda之两个向量点乘
cuda之两个向量点乘
断桥残雪断桥残雪 发表于2年前
cuda之两个向量点乘
  • 发表于 2年前
  • 阅读 70
  • 收藏 0
  • 点赞 1
  • 评论 0

腾讯云 技术升级10大核心产品年终让利>>>   

摘要: 本文主要用代码演示了两个向量点乘的cuda并行计算代码。

1.简介

本文中主要介绍了使用block来集散,每个block计算数组上对应两个数的乘积,最后将结果数组拷贝到CPU主内存上即可。

2.cuda代码演示


#include<iostream>
#include<cstdlib>
using namespace std;
#define N 100
__global__ void add(int *a,int *b,int *c) {
	int ind = blockIdx.x;
	c[ind] = a[ind] * b[ind];
}

void initialVector(int length,int *arr,int limit) {
	for(int i=0;i<length;i++,limit++)
		*(arr+i) = limit;
}
int main() {
	int *a,*b,*c;
	a = (int *)malloc(sizeof(int)*N);
	b = (int *)malloc(sizeof(int)*N);
	c = (int *)malloc(sizeof(int)*N);
	initialVector(N,a,1);
	initialVector(N,b,1);
	int *d_a,*d_b,*d_c;
        /* allocate memory on device */
	cudaMalloc((void **)&d_a,N*sizeof(int));	
	cudaMalloc((void **)&d_b,N*sizeof(int));	
	cudaMalloc((void **)&d_c,N*sizeof(int));	
	cudaMemcpy(d_a,a,N*sizeof(int),cudaMemcpyHostToDevice);
	cudaMemcpy(d_b,b,N*sizeof(int),cudaMemcpyHostToDevice);
	add<<<N,1>>>(d_a,d_b,d_c);
	cudaMemcpy(c,d_c,N*sizeof(int),cudaMemcpyDeviceToHost);
	cout<<"c["<<N-1<<"]:"<<c[N-1]<<endl;
	free(a);
	free(b);
	free(c);
	cudaFree(d_a);
	cudaFree(d_b);
	cudaFree(d_c);
	return 0;
}




共有 人打赏支持
粉丝 50
博文 125
码字总数 94909
×
断桥残雪断桥残雪
如果觉得我的文章对您有用,请随意打赏。您的支持将鼓励我继续创作!
* 金额(元)
¥1 ¥5 ¥10 ¥20 其他金额
打赏人
留言
* 支付类型
微信扫码支付
打赏金额:
已支付成功
打赏金额: