您可以拥有任意数量的共享内存声明。然而,运行时只分配一个共享内存缓冲区,每个共享内存数组将被分配相同的地址(即共享内存分配的起始地址)。因此,例如:
#include <cstdio>
extern __shared__ int a[];
extern __shared__ int b[];
extern __shared__ int c[];
__global__
void kernel(void)
{
int * a0 = &a[0];
int * b0 = &b[0];
int * c0 = &c[0];
printf("a0 = %#x \n", a0);
printf("b0 = %#x \n", b0);
printf("c0 = %#x \n", c0);
}
int main()
{
kernel<<<1,1,1024>>>();
cudaDeviceReset();
return 0;
}
这个做什么:
$ nvcc -arch=sm_30 -run extshm.cu
a0 = 0x1000000
b0 = 0x1000000
c0 = 0x1000000
如果想要有两个共享数组,在任何支持(即计算能力>=2.0)的GPU上,可以像这样做:
#include <cstdio>
extern __shared__ int a[];
__global__
void kernel(void)
{
int * a0 = &a[0];
int * b0 = &a[1024];
int * c0 = &a[1024+768];
printf("a0 = %#x \n", a0);
printf("b0 = %#x \n", b0);
printf("c0 = %#x \n", c0);
}
int main()
{
kernel<<<1,1,1024+768+512>>>();
cudaDeviceReset();
return 0;
}
这将会得到:
nvcc -arch=sm_30 -run extshm2.cu
a0 = 0x1000000
b0 = 0x1001000
c0 = 0x1001c00
我认为后者是您正在寻找的内容。