CUDA是否提供类似std::floating_point且包含所有IEEE754类型（如__half）的concept？

CUDA官方并没有提供和std::floating_point完全对应、且包含__half这类CUDA扩展IEEE754浮点类型的标准Concept。std::floating_point仅识别C++标准库定义的浮点类型（float、double、long double），而__half是CUDA特有的16位浮点扩展类型，不在标准库的浮点类型范畴内，这就是你提供的代码中use_floating_point<__half>()触发编译错误的原因。

不过我们可以通过两种方式自定义一个覆盖所有CUDA支持的IEEE754浮点类型的Concept：

方式一：显式枚举CUDA扩展类型 #

直接将标准浮点类型和CUDA特有的浮点类型纳入判断逻辑：

#include <concepts>
#include <type_traits>
#include <cuda_fp16.h>
#include <cuda_bf16.h>

template<typename T>
concept cuda_floating_point = 
    std::floating_point<T> || 
    std::is_same_v<T, __half> || 
    std::is_same_v<T, __bfloat16>;

方式二：利用CUDA内置类型判断（更通用） #

CUDA编译器提供了内置的__is_floating_point类型 trait，它能识别所有CUDA支持的浮点类型（包括标准浮点类型、__half、__bfloat16等），用它来实现Concept更简洁通用：

#include <concepts>
#include <cuda_fp16.h>

template<typename T>
concept cuda_floating_point = __is_floating_point(T);

修改后的测试代码 #

用自定义的cuda_floating_point替换原代码中的std::floating_point，即可让__half正常适配：

#include <concepts>
#include <iostream>
#include <cuda_fp16.h>

template<cuda_floating_point T>
__host__ __device__ bool use_floating_point() {
    return true; // 注意原代码中的True是拼写错误，C++关键字为小写true
}

int main() {
  std::cout << "Is float a floating-point type? " << std::boolalpha << use_floating_point<float>() << '\n';
  std::cout << "Is __half a floating-point type? " << std::boolalpha << use_floating_point<__half>() << '\n'; // 无编译错误
  return 0;
}

注意：如果使用CUDA 11.0及以上版本，也可以尝试使用<cuda/std/concepts>中的CUDA标准库扩展，但直接依赖编译器内置的__is_floating_point是兼容性最好的方案。

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