CUDA编程学习(四)
2024-08-29 19:59:27
利用Block和Thread进行并行加速
_global_ void add(int *a, int *b, int *c)
{
int index = threadIdx.x + blockIdx.x * blockDim.x;
c[index] = a[index] + b[index];
} #define N (2048*2048)
#define THREAD_PER_BLOCK 512 int main()
{
int *a, *b, *c; //host copies of a, b, c
int *d_a, *d_b, *d_c; //device copies of a, b, c
int size = N * sizeof(int); //Alloc space for device copies of a, b, c
cudaMalloc((void **)&d_a, size);
cudaMalloc((void **)&d_b, size);
cudaMalloc((void **)&d_c, size); //Alloc space for host copies of a, b, c and setup input values
a = (int *)malloc(size); random_ints(a, N);
b = (int *)malloc(size); random_ints(b, N);
c = (int *)malloc(size); //Copy the data into device
cudeMemcpy(d_a, a, size, cudaMemcpyHostToDevice);
cudaMemcpy(d_b, b, size, cudaMemcpyHostToDevice); //Launch add() kernel on GPU with N blocks
add<<<N/THREADS_PER_BLOCK,THREADS_PER_BLOCK>>>(d_a, d_b, d_c); //Copy result back to host
cudaMemcpy(c, d_c, size, cudaMemcpyDeviceToHost); //Cleanup
free(a); free(b); free(c);
cudeFree(d_a); cudaFree(d_b); cudaFree(d_c);
return ;
} /**** What's the function of random_ints****/
void random_ints(int* a, int N)
{
int i;
for (i = ; i < N; ++i)
a[i] = rand();
}
最新文章
- 进程管理supervisor的简单说明
- 设置 tableview 的背景颜色,总是有蒙层
- textarea关于空格和换行那点事
- bug--常见的bug总结:
- CentOS-6.5安装zabbix 3.0.4
- JS Jquery去除数组重复元素
- 一、C# 概述
- [转载] 50个Android开发人员必备UI效果源码
- 使用CSS如何悬停背景颜色变色 onmouseover、onmouseout
- Eclipse4JavaEE安装Gradle,并导入我们的Gradle项目
- java或Jmeter实现两个日期相加减(2003-06-01-2003-05-01)
- font-spider-plus,字体压缩插件使用笔记
- Python元组的一点用法
- Listen and Write 18th Feb 2019
- MTK6261 11C之Init Analysis【手记】
- Android应用内嵌cocos2dx游戏项目
- Linux tree命令详解
- Linux alias命令详解
- Java基础——JVM内存结构
- Openwrt之移动硬盘ext3/ext4格式化工具
热门文章
- jquery实践案例--验证电子邮箱
- 烂泥:centos安装LVM方式
- .net framework 4.6.2 下载
- Windows Live Writer离线编写博客
- Oracle 分组聚合二种写法,listagg和wmsys.wm_concat
- IIS关于“ 配置错误 不能在此路径中使用此配置节”的解决办法
- ifrog-1028 Bob and Alice are playing numbers(trie树)
- HDU 4082 Hou Yi&#39;s secret --枚举
- ZOJ 2301 / HDU 1199 Color the Ball 离散化+线段树区间连续最大和
- HTML设置超链接字体颜色和点击后的字体颜色