gpt4 book ai didi

cuda - 在 CUDA 中分配共享变量

转载 作者:行者123 更新时间:2023-12-02 07:05:53 26 4
gpt4 key购买 nike

如何在CUDA中分配共享变量?我有一个内核,需要在属于特定 block 的线程之间共享数据。我需要两个名为 sideid 的共享变量。我是这样使用的:

extern __shared__ int sid, eid  

但它给了我一个错误,即 __shared__ 变量不能有外部链接。

最佳答案

共享内存有两种分配方式:静态和动态

1、静态

  __shared__ int Var1[10]

2、动态:应添加“extern”关键字

extern __shared__ int Var1[]

如果您使用动态方式分配共享内存,您应该在调用函数时设置共享内存大小。例如:
testKernel <<< grid, threads, size>>>(...)
第三段是共享内存的大小。这样所有的共享内存都从同一个地址开始。所以如果你想定义几个共享的内存。您应该编写如下代码。

__global__ void func(...)
{
extern __shared__ char array[];
short * array0 = (short*)array;
float * array1 = (float*)(&array0[128]);
}

关于cuda - 在 CUDA 中分配共享变量,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/12066730/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com