cuda：thread->block->stream

weixin_30713953

于 2016-05-19 09:54:00 发布

阅读量83

点赞数

CC 4.0 BY-SA版权

文章标签： python c/c++

原文链接：http://www.cnblogs.com/linyuanzhou/p/5507671.html

本文介绍了CUDA编程的基本概念，包括核函数的定义与调用方法，以及如何利用线程结构进行矩阵和向量运算。

程序结构

1.核函数

核函数的定义和c语言方式类似，使用__global__什么核函数，线程的数目通过<<<...,nums>>>来传递。

// Kernel definition
__global__ void VecAdd(float* A, float* B, float* C)
{
    int i = threadIdx.x;
    C[i] = A[i] + B[i];
}
int main()
{
    ...
    // Kernel invocation with N threads
    VecAdd<<<1, N>>>(A, B, C);
    ...
}

2.线程的结构

线程是一个三维向量（x,y,z)，在使用的过程中，可以使用(x),(x,y),(x,y,z)

以下，是一个使用二维（x,y)的核函数

// Kernel definition
__global__ void MatAdd(float A[N][N], float B[N][N],
                       float C[N][N])
{
    int i = threadIdx.x;
    int j = threadIdx.y;
    C[i][j] = A[i][j] + B[i][j];
}
int main()
{
    ...
    // Kernel invocation with one block of N * N * 1 threads
    int numBlocks = 1;
    dim3 threadsPerBlock(N, N);
    MatAdd<<<numBlocks, threadsPerBlock>>>(A, B, C);
    ...
}