CUDA 11.8编译16位无符号整数atomicCAS失败，文档是否有误？

核心原因 #

CUDA中unsigned short类型的atomicCAS并非在所有GPU计算架构（compute capability）下都支持。即便CUDA Toolkit 11.8文档标注该函数受支持，也需要显式指定支持16位原子操作的计算架构，才能让nvcc识别并编译该函数。

CUDA 11.8里，unsigned short版本的atomicCAS仅对compute capability ≥ 6.0（即Pascal架构及后续型号）的GPU生效。如果编译时未指定符合要求的架构，nvcc会默认使用较低的兼容架构（比如sm_30），这类旧架构不支持16位原子操作，因此会触发函数匹配失败的报错。

解决步骤 #

1. 确认目标GPU的计算架构：通过nvidia-smi查看GPU型号，对应到其compute capability（例如RTX 20系列对应sm_75，RTX 30系列对应sm_86）。
2. 编译时指定架构参数：在nvcc编译命令中添加-arch=sm_XX参数，XX为对应架构的版本号。示例：

   nvcc -arch=sm_60 hello.cu -o hello
   

   若需兼容多个架构，可使用-gencode参数，示例：

   nvcc -gencode arch=compute_60,code=sm_60 -gencode arch=compute_75,code=sm_75 hello.cu -o hello
   

补充说明 #

官方文档不存在错误，文档中注明的“支持”是指在符合条件的计算架构下有效，而非所有默认编译场景都支持。未指定架构时，nvcc会采用向下兼容的默认设置，导致无法识别16位原子操作函数。

内容的提问来源于stack exchange，提问作者MaiaVictor