CUDA 二维矩阵操作
2016-05-05 01:20
441 查看
二维矩阵
代码如下:
#include <stdio.h>
#include <stdlib.h>
#include"cuda_runtime.h"
#include"device_launch_parameters.h"
#include"iostream"
using namespace std;
/************************************************************************/
/* Example:矩阵加法 */
const int N =10;
__global__ void add(float m[]
, float n[]
);
void main()
{
float(*a)
= new float
;
float(*b)
= new float
;
float(*dev_a)
, (*dev_b)
;
cout << "a:" << endl;
for (int i = 0; i < N; i++)
{
for (int j = 0; j < N; j++)
{
a[i][j] = i + j;
cout << a[i][j] << " ";
}
cout << endl;
}
cudaMalloc((void **)&dev_a, N * N * sizeof(float));
cudaMalloc((void **)&dev_b, N * N * sizeof(float));
cudaMemcpy(dev_a, a, N * N * sizeof(float), cudaMemcpyHostToDevice);
// delete[]a;
dim3 blocks(N, N);
dim3 threads(1);
add << <blocks, threads >> >(dev_a, dev_b);
cudaMemcpy(b, dev_b, N * N * sizeof(float), cudaMemcpyDeviceToHost);
cudaFree(dev_a);
cudaFree(dev_b);
cout << "b:" << endl;
for (int i = 0; i < N; i++)
{
for (int j = 0; j < N; j++)
{
cout << b[i][j] << " ";
}
cout << endl;
}
delete[]b;
}
__global__ void add(float m[]
, float n[]
)
{
int x = threadIdx.x + blockIdx.x * blockDim.x;
int y = threadIdx.y + blockIdx.y * blockDim.y;
{
n[x][y] = 200*m[x][y];
}
}
代码如下:
#include <stdio.h>
#include <stdlib.h>
#include"cuda_runtime.h"
#include"device_launch_parameters.h"
#include"iostream"
using namespace std;
/************************************************************************/
/* Example:矩阵加法 */
const int N =10;
__global__ void add(float m[]
, float n[]
);
void main()
{
float(*a)
= new float
;
float(*b)
= new float
;
float(*dev_a)
, (*dev_b)
;
cout << "a:" << endl;
for (int i = 0; i < N; i++)
{
for (int j = 0; j < N; j++)
{
a[i][j] = i + j;
cout << a[i][j] << " ";
}
cout << endl;
}
cudaMalloc((void **)&dev_a, N * N * sizeof(float));
cudaMalloc((void **)&dev_b, N * N * sizeof(float));
cudaMemcpy(dev_a, a, N * N * sizeof(float), cudaMemcpyHostToDevice);
// delete[]a;
dim3 blocks(N, N);
dim3 threads(1);
add << <blocks, threads >> >(dev_a, dev_b);
cudaMemcpy(b, dev_b, N * N * sizeof(float), cudaMemcpyDeviceToHost);
cudaFree(dev_a);
cudaFree(dev_b);
cout << "b:" << endl;
for (int i = 0; i < N; i++)
{
for (int j = 0; j < N; j++)
{
cout << b[i][j] << " ";
}
cout << endl;
}
delete[]b;
}
__global__ void add(float m[]
, float n[]
)
{
int x = threadIdx.x + blockIdx.x * blockDim.x;
int y = threadIdx.y + blockIdx.y * blockDim.y;
{
n[x][y] = 200*m[x][y];
}
}
相关文章推荐
- 使用C++实现JNI接口需要注意的事项
- 关于指针的一些事情
- c++ primer 第五版 笔记前言
- share_ptr的几个注意点
- Lua中调用C++函数示例
- Lua教程(一):在C++中嵌入Lua脚本
- Lua教程(二):C++和Lua相互传递数据示例
- C++联合体转换成C#结构的实现方法
- C++高级程序员成长之路
- C++编写简单的打靶游戏
- C++ 自定义控件的移植问题
- C++变位词问题分析
- C/C++数据对齐详细解析
- C++基于栈实现铁轨问题
- C++中引用的使用总结
- 使用Lua来扩展C++程序的方法
- C++中调用Lua函数实例
- Lua和C++的通信流程代码实例
- C与C++之间相互调用实例方法讲解
- 解析C++中派生的概念以及派生类成员的访问属性