关于CUDA宏__CUDA_ARCH__的问题

Question

我在 ttt.cu 中有一个简单的ttt.cu代码

#include <iostream>
__global__ void example(){
   printf("__CUDA_ARCH__: %d \n", __CUDA_ARCH__);
}
int main(){
example<<<1,1>>>();
}

使用 CMakeLists.txt：

cmake_minimum_required(VERSION 3.18)
project(Hello)
find_package(CUDA REQUIRED)

cuda_add_executable(sss ttt.cu)

然后我得到了错误： identifier "__CUDA_ARCH__" is undefined 。 我想知道为什么会发生这种情况，我应该怎么做才能使__CUDA_ARCH__有效？ 我们可以在 header .h文件中的主机代码中使用有效的__CUDA_ARCH__吗？

Answer 1

__CUDA_ARCH__是一个编译器宏。

我们可以在主机代码中使用有效的__CUDA_ARCH__

不，它仅用于设备代码：

主机代码（非 GPU 代码）不能依赖它。

您无法按照您想象的方式打印编译器宏。 它不是 C++ 中定义的普通数值变量。 你可以做这样的事情， 但这会在编译时打印，而不是在运行时打印。

要在运行时打印，您可以执行以下操作：

$ cat t2.cu
#include <cstdio>
#define STR_HELPER(x) #x
#define STR(x) STR_HELPER(x)

__device__ void print_arch(){
  const char my_compile_time_arch[] = STR(__CUDA_ARCH__);
  printf("__CUDA_ARCH__: %s\n", my_compile_time_arch);
}
__global__ void example()
{
   print_arch();
}

int main(){

example<<<1,1>>>();
cudaDeviceSynchronize();
}
$ nvcc -o t2 t2.cu

$ ./t2
__CUDA_ARCH__: 520
$

请注意，这里有很多关于cuda标签讨论__CUDA_ARCH__的问题，您可能希望查看其中的一些问题。