CUDA 常量内存值不正确 [重复]

Question

我一直在阅读许多与恒定记忆相关的 SO 问题，但我仍然不明白为什么我的程序无法正常工作。总体看起来如下

Common.cuh

__constant__ int numElements;

__global__
void kernelFunction();

Common.cu

#include "Common.cuh"
#include <stdio.h>

__global__
kernelFunction()
{
   printf("NumElements = %d", numElements);
}

Test.cu

#include "Common.cuh"

int main()
{
   int N = 100;
   cudaMemcpyToSymbol(numElements,&N,sizeof(int));
   kernelFunction<<<1,1>>>();
   cudaDeviceSynchronize();
   return 0;
}

它编译时没有错误，但是当打印numElements 的值时，我只是得到一个随机值。有人可以指出我正确的方向来理解这一点吗？

Answer 1

这一行：

__constant__ int numElements;

具有编译单元范围。这意味着如果你将它编译到一个模块中，也编译到另一个模块中，这两个模块将在__constant__ 内存中具有不同的numElements 实例化。

解决方案是使用separate compilation and linking，将两个模块设备链接在一起，此时设备链接器将在两个模块之间解析符号。

nvcc -arch=sm_20 -rdc=true -o test common.cu test.cu

示例：

$ cat common.cuh
#ifndef COMMON_CU
extern __constant__ int numElements;
#endif
__global__
void kernelFunction();
$ cat common.cu
#define COMMON_CU
#include "common.cuh"
#include <stdio.h>

__constant__ int numElements;
__global__
void kernelFunction()
{
   printf("NumElements = %d\n", numElements);
}
$ cat test.cu
#define TEST_CU
#include "common.cuh"

int main()
{
   int N = 100;
   cudaMemcpyToSymbol(numElements,&N,sizeof(int));
   kernelFunction<<<1,1>>>();
   cudaDeviceSynchronize();
   return 0;
}

$ nvcc -arch=sm_20 -rdc=true -o test common.cu test.cu
$ ./test
NumElements = 100
$