cuda共享内存(静态和动态)的使用和注意点

共享内存(shared memory)
● 比global快上百倍
● 可以通过缓存数据减少对global memory的读写
● 线程间可以使用shared memory进行协作

共享内存会被分为32个bank,和warp中的线程数一致,即一个线程访问一个bank

从软件概念来讲,一个共享内存是在一个block内,block内的所有线程都能使用该共享内存

从硬件概念来讲,一个共享内存是在一个sm中

在使用共享内存时,一般会有两种用法:

静态共享内存

声明方式:
● __shared__关键字,固定大小
__shared__ float a[N];
这种使用方式可以定义多个共享内存变量,他们之间不会冲突,可以分别使用

动态共享内存

声明方式:
● extern __shared__关键字,不固定大小,大小由调用核函数<<<… shared_size>>>时指定。
extern __shared__ float a[];
这种使用方式原理是这样的:
首先注明了c++ 关键词 extern,表示该处只是声明而不是定义,说明这个变量不是在写extern shared float a[];这里定义的;
那是在哪里定义的呢?其实是cuda runtime 根据kernel函数插入的参数隐式的定义的;
那我们怎么知道它定义的地址呢?就是使用声明来声明一个变量就行,在一个核函数内,可以声明多个名字不同的extern __shared变量,但是实际上它们都是指向的同一个内存地址;
比如在代码中写了:
extern __shared__ float a[];
extern __shared__ float b[];
编译不会报错,你以为这是两个变量,但是实际上a和b是同一块内存,都是cuda runtime 为这个核函数定义的那块共享内存,a和b只是对这块内存的不同的名字而已
如果想使用多个数组,可以用指针,
在这里插入图片描述
每个thread block的最大shared memory大小为64KB(arch-specific)
如果想声明超过48KB的共享内存,必须使用dynamic shared memory 且要设置cudaFuncAttributeMaxDynamicSharedMemorySize

  • 9
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
CUDA共享内存是一种位于GPU上的高速存,可以用于在同一个线程块内的线程之间共享数据。使用内存可以显著提内存访问的效,并且减少对全局内存访问次数。 以下使用CUDA共享内存的一般步: 1. 声明共内存:在GPU核函数中,可以使用`__shared__`关键字来声明共享内存共享内存的大小需要在编译时确定,并且是所有线程块中的线程共享的。 ```cuda __shared__ float sharedData[SIZE]; ``` 2. 将数据从全局内存复制到共享内存:在GPU核函数中,使用线程块中的线程来将数据从全局内存复制到共享内存中。可以使用线程块索引和线程索引来确定数据的位置。 ```cuda int tid = threadIdx.x; int blockId = blockIdx.x; int index = blockId * blockDim.x + tid; sharedData[tid] = globalData[index]; ``` 3. 同步线程块:在将数据复制到共享内存后,需要使用`__syncthreads()`函数来同步线程块中的线程。这样可以确保所有线程都已经将数据复制到共享内存中。 ```cuda __syncthreads(); ``` 4. 使用共享内存:一旦所有线程都已经将数据复制到共享内存中,可以使用共享内存进行计算。由于共享内存位于GPU的高速缓存中,所以访问速度较快。 ```cuda sharedData[tid] += 1.0f; ``` 5. 将数据从共享内存复制回全局内存:在计算完成后,可以使用线程块中的线程将数据从共享内存复制回全局内存。 ```cuda globalData[index] = sharedData[tid]; ``` 需要注意的是,共享内存的大小是有限的,不同的GPU架构及型号都有不同的限制。因此,在使用共享内存时,需要确保不超过设备的限制,并且合理地利用共享内存,以提高性能。此外,需要注意线程同步的位置和使用方法,以避免数据竞争和错误的结果。 以上是使用CUDA共享内存的基本步骤,具体的实现方式会根据具体问题而有所不同。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值