CUDA Programming GuideCUDA 编程指南中文图解
首页
阅读地图
1.1 Introduction
PDF 图版
  • 源代码仓库(GitHub)
  • CUDA Programming Guide Release 13.3
  • NVIDIA 官方在线目录
  • 本地原始 PDF
首页
阅读地图
1.1 Introduction
PDF 图版
  • 源代码仓库(GitHub)
  • CUDA Programming Guide Release 13.3
  • NVIDIA 官方在线目录
  • 本地原始 PDF
  • 5. Technical Appendices

    • 5.1 计算能力(Compute Capabilities)
    • 5.2 CUDA 环境变量(CUDA Environment Variables)
    • 5.3 C++ 语言支持(C++ Language Support)
    • 5.4 C/C++ 语言扩展(C/C++ Language Extensions)
    • 5.5 浮点计算(Floating-Point Computation)
    • 5.6 设备可调用 API 与内建函数(Device-Callable APIs and Intrinsics)
    • 5.7 CUDA C++ 内存模型(CUDA C++ Memory Model)
    • 5.8 CUDA C++ 执行模型(CUDA C++ Execution model)

本页按 NVIDIA CUDA Programming Guide 5.5 节整理,保留官方段落、列表、表格、代码、图示和交叉引用。

原文:5.5. Floating-Point Computation

5.5. 浮点计算(Floating-Point Computation)

5.5.1. 浮点计算简介(Floating-Point Introduction)

自1985年采用IEEE-754二进制浮点算术标准以来,几乎所有主流计算系统,包括 NVIDIA 的 CUDA 架构,都已实现了该标准。IEEE-754 标准规定了浮点算术结果应如何近似。

为了获得准确的结果并以所需的精度实现最高性能,考虑浮点行为的许多方面非常重要。这在异构计算环境中尤为重要,因为操作是在不同类型的硬件上执行的。

以下部分回顾了浮点计算的基本属性,并涵盖了融合乘加(FMA)运算和点积。这些示例说明了不同的实现选择如何影响精度。

5.5.1.1. 浮点格式(Floating-Point Format)

浮点格式和功能在IEEE-754标准中定义。

该标准规定二进制浮点数据由三个字段编码:

  • 符号位:一位,用于指示正数或负数。
  • 指数:编码以数值偏置偏移的以2为底的指数。
  • 有效数(也称为尾数或小数部分):编码数字的小数值。

浮点编码

最新的 IEEE-754 标准定义了以下二进制格式的编码和属性:

  • 16位,也称为半精度,对应于 CUDA 中的 __half 数据类型。
  • 32位,也称为单精度,对应于 C、C++ 和 CUDA 中的 float 数据类型。
  • 64位,也称为双精度,对应于 C、C++ 和 CUDA 中的 double 数据类型。
  • 128位,也称为四倍精度,对应于 CUDA 中的 __float128 或 _Float128 数据类型。

这些类型具有以下位长度:

IEEE-754 浮点编码

与规约数的浮点编码相关的数值计算如下:

[(-1)^\mathrm{sign} \times 1.\mathrm{mantissa} \times 2^{\mathrm{exponent} - \mathrm{bias}}]

对于非规约数,公式修改为:

[(-1)^\mathrm{sign} \times 0.\mathrm{mantissa} \times 2^{1-\mathrm{bias}}]

对于单精度和双精度,指数分别偏置了 (127) 和 (1023)。(1.) 的整数部分隐含在分数中。

例如,值 (-192 = (-1)^1 \times 2^7 \times 1.5) 被编码为负号、指数 (7) 和小数部分 (0.5)。因此,对于 float,指数 (7) 由值为 7 + 127 = 134 = 10000110 的位串表示;对于 double,由 7 + 1023 = 1030 = 10000000110 表示。尾数 0.5 = 2^-1 由二进制值表示,其第一位为 1。单精度和双精度下 (-192) 的二进制编码如下图所示:

 的浮点数表示

由于小数部分字段使用的位数有限,并非所有实数都能被精确表示。例如,分数 (2 / 3) 的数学值的二进制表示为 0.10101010...,其在二进制小数点后有无限多位。因此,(2 / 3) 在被表示为有限精度的浮点数之前必须进行舍入。舍入规则和模式在 IEEE-754 标准中规定。最常用的模式是舍入到最近偶数,简称舍入到最近。

5.5.1.2. 规格化值与非规格化值(Normal and Subnormal Values)

任何指数字段既不全为零也不全为一的浮点数值都称为规格化值。

浮点数值的一个重要方面是,最小的可表示正规格化数 FLT_MIN 与零之间存在巨大的间隔。这个间隔远大于 FLT_MIN 与第二小的规格化数之间的间隔。

浮点非规格化数,也称为次正规数,正是为了解决这个问题而引入的。一个非规格化浮点数值的指数部分所有位都设置为零,并且有效数字部分至少有一位被设置。非规格化数是 IEEE-754 浮点标准的必要组成部分。

非规格化数允许精度逐渐损失,而不是突然向零舍入。然而,非规格化数的计算成本更高。因此,不需要严格精度的应用程序可能会选择避免使用它们以提高性能。nvcc 编译器允许通过设置 -ftz=true 选项(刷新到零)来禁用非规格化数,该选项也包含在 --use_fast_math 中。

单精度下最小规格化值和非规格化值的编码简化示意图如下:

最小规格化值与非规格化值表示

其中 X 代表 0 和 1。

5.5.1.3. 特殊值(Special Values)

IEEE-754 标准为浮点数定义了三种特殊值:

零:

  • 数学上的零。
  • 注意,浮点零有两种可能的表示:+0 和 -0。这与整数零的表示不同。
  • +0 == -0 的计算结果为 true。
  • 零的编码是指数和有效数字的所有位都设置为 0。

无穷大:

  • 浮点数遵循饱和算术规则,即运算结果超出可表示范围时会产生 +Infinity 或 -Infinity。
  • 无穷大的编码是指数部分所有位设置为 1,有效数字部分所有位设置为 0。无穷大值恰好有两种编码。
  • 涉及无穷大和有限非零值的算术运算通常会产生无穷大。不确定形式如 Inf * 0.0、Inf - Inf、Inf / Inf 和 0.0 / 0.0 会产生 NaN。

非数(NaN):

  • NaN 是一个特殊的符号,表示未定义或无法表示的值。常见的例子有 0.0 / 0.0、sqrt(-1.0) 或 +Inf - Inf。
  • NaN 的编码是指数部分的所有位都设置为 1,而有效数字部分可以是除全 0 之外的任何位模式。共有 (2^{\mathrm{mantissa} + 1} - 2) 种可能的编码。
  • 任何涉及 NaN 的算术运算都会产生 NaN。
  • 任何涉及 NaN 的有序比较(<、<=、>、>=、==)都会返回 false,包括 NaN == NaN(非自反性)。无序比较 NaN != NaN 返回 true。
  • NaN 以两种形式提供:静默 NaN(qNaN)用于传播由无效操作或值引起的错误。无效的算术运算通常会产生静默 NaN。它们的编码是有效数字的最高有效位设置为 1。信号 NaN(sNaN)旨在引发无效操作异常。信号 NaN 通常是显式创建的。它们的编码是有效数字的最高有效位设置为 0。静默 NaN 和信号 NaN 的确切位模式由具体实现定义。CUDA 提供了 cuda::std::numeric_limits<T>::quiet_NaN 和 cuda::std::numeric_limits<T>::signaling_NaN 常量来获取它们的特殊值。

特殊值编码的简化可视化如下图所示:

无穷大和 NaN 的浮点数表示

其中 X 代表 0 和 1。

5.5.1.4. 结合律(Associativity)

需要注意的是,由于浮点运算精度有限,数学算术的规则和性质不能直接应用于浮点运算。下面的例子展示了单精度值 A、B 和 C,以及使用不同结合律计算出的它们之和的精确数学值。

[\begin{split}\begin{aligned} A &= 2^{1} \times 1.00000000000000000000001 \ B &= 2^{0} \times 1.00000000000000000000001 \ C &= 2^{3} \times 1.00000000000000000000001 \ (A + B) + C &= 2^{3} \times 1.01100000000000000000001011 \ A + (B + C) &= 2^{3} \times 1.01100000000000000000001011 \end{aligned}\end{split}]

在数学上,((A + B) + C) 等于 (A + (B + C))。

令 (\mathrm{rn}(x)) 表示对 (x) 进行一次舍入操作。根据 IEEE-754 标准,在单精度浮点运算中采用最近舍入模式执行相同的计算,我们得到:

[\begin{split}\begin{aligned} A + B &= 2^{1} \times 1.1000000000000000000000110000\ldots \ \mathrm{rn}(A+B) &= 2^{1} \times 1.10000000000000000000010 \ B + C &= 2^{3} \times 1.0010000000000000000000100100\ldots \ \mathrm{rn}(B+C) &= 2^{3} \times 1.00100000000000000000001 \ A + B + C &= 2^{3} \times 1.0110000000000000000000101100\ldots \ \mathrm{rn}\big(\mathrm{rn}(A+B) + C\big) &= 2^{3} \times 1.01100000000000000000010 \ \mathrm{rn}\big(A + \mathrm{rn}(B+C)\big) &= 2^{3} \times 1.01100000000000000000001 \end{aligned}\end{split}]

作为参考,上面也计算了精确的数学结果。根据 IEEE-754 计算的结果与精确的数学结果不同。此外,对应于求和 (\mathrm{rn}(\mathrm{rn}(A + B) + C)) 和 (\mathrm{rn}(A + \mathrm{rn}(B + C))) 的结果也彼此不同。在这种情况下,(\mathrm{rn}(A + \mathrm{rn}(B + C))) 比 (\mathrm{rn}(\mathrm{rn}(A + B) + C)) 更接近正确的数学结果。

这个例子表明,即使所有基本操作都符合 IEEE-754,看似相同的计算也可能产生不同的结果。

5.5.1.5. 融合乘加(FMA)(Fused Multiply-Add (FMA))

融合乘加 (FMA) 操作仅通过一次舍入步骤计算结果。如果没有 FMA,结果将需要两次舍入步骤:一次用于乘法,一次用于加法。因为 FMA 只使用一次舍入步骤,所以它能产生更准确的结果。

融合乘加操作对 NaN 传播的影响可能与两个独立操作不同。然而,FMA 的 NaN 处理并非在所有目标平台上都完全相同。对于具有多个 NaN 操作数的不同实现,可能倾向于选择静默 NaN 或传播某个操作数的有效载荷。此外,当存在多个 NaN 操作数时,IEEE-754 并未严格规定确定性的有效载荷选择顺序。NaN 也可能出现在中间计算中,例如 (\infty \times 0 + 1) 或 (1 \times \infty - \infty),从而产生一个由实现定义的 NaN 有效载荷。


为了清晰起见,首先考虑一个使用十进制算术的例子来说明 FMA 操作的工作原理。我们将使用总共五位精度(小数点后四位)来计算 (x^2 - 1)。

  • 对于 (x = 1.0008),正确的数学结果是 (x^2 - 1 = 1.60064 \times 10^{-4})。仅使用小数点后四位的最接近数字是 (1.6006 \times 10^{-4})。
  • 融合乘加操作仅通过一次舍入步骤即可获得正确结果:(\mathrm{rn}(x \times x - 1) = 1.6006 \times 10^{-4})。
  • 另一种方法是分别计算乘法和加法步骤。(x^2 = 1.00160064) 转换为 (\mathrm{rn}(x \times x) = 1.0016)。最终结果是 (\mathrm{rn}(\mathrm{rn}(x \times x) -1) = 1.6000 \times 10^{-4})。

分别对乘法和加法进行舍入得到的结果偏差为 (0.00064)。相应的 FMA 计算仅偏差 (0.00004),其结果最接近正确的数学答案。结果总结如下:

[\begin{split}\begin{aligned} x &= 1.0008 \ x^{2} &= 1.00160064 \ x^{2} - 1 &= 1.60064 \times 10^{-4} && \text{真实值} \ \mathrm{rn}\big(x^{2} - 1\big) &= 1.6006 \times 10^{-4} && \text{融合乘加} \ \mathrm{rn}\big(x^{2}\big) &= 1.0016 \ \mathrm{rn}\big(\mathrm{rn}(x^{2}) - 1\big) &= 1.6000 \times 10^{-4} && \text{先乘后加} \end{aligned}\end{split}]


下面是另一个使用二进制单精度值的示例:

[\begin{split}\begin{aligned} A &= 2^{0} \times 1.00000000000000000000001 \ B &= -2^{0} \times 1.00000000000000000000010 \ \mathrm{rn}\big(A \times A + B\big) &= 2^{-46} \times 1.00000000000000000000000 && \text{融合乘加} \ \mathrm{rn}\big(\mathrm{rn}(A \times A) + B\big) &= 0 && \text{先乘后加} \end{aligned}\end{split}]

  • 分别计算乘法和加法会导致所有精度位丢失,结果为 (0)。
  • 而计算 FMA 则能提供等于数学值的结果。

融合乘加有助于防止在相消性减法中损失精度。当符号相反、量级相近的数相加时,就会发生相消性减法。在这种情况下,许多高位会相互抵消,导致有意义的位数减少。融合乘加在乘法过程中计算双倍宽度的乘积。因此,即使在加法过程中发生相消性减法,乘积中仍有足够的有效位来产生精确的结果。


CUDA 中的融合乘加支持:

CUDA 为 float 和 double 数据类型提供了多种实现融合乘加操作的方式:

  • 使用 -fmad=true 或 --use_fast_math 标志编译时的 x * y + z 表达式。
  • C 标准库函数 fma(x, y, z) 和 fmaf(x, y, z)。
  • CUDA 数学内置函数 __fmaf_[rd, rn, ru, rz]、__fmaf_ieee_[rd, rn, ru, rz] 和 __fma_[rd, rn, ru, rz]。
  • CUDA C++ 标准库函数 cuda::std::fma(x, y, z) 和 cuda::std::fmaf(x, y, z)。

主机平台上的融合乘加支持:

是否使用融合操作取决于平台对该操作的支持情况以及代码的编译方式。在比较 CPU 和 GPU 结果时,了解主机平台对融合乘加的支持非常重要。

  • 编译器标志和融合乘加硬件支持:GCC 和 Clang 的 -mfma、NVC++ 的 -Mfma 以及 Microsoft Visual Studio 的 /fp:contract。例如,对于具有 AVX2 ISA 的 x86 平台,使用 GCC 或 Clang 的 -mavx2 标志以及 Microsoft Visual Studio 的 /arch:AVX2 编译的代码。对于具有高级 SIMD (Neon) ISA 的 Arm64 (AArch64) 平台。
  • C 标准库函数 fma(x, y, z) 和 fmaf(x, y, z)。
  • C++ 标准库函数 std::fma(x, y, z) 和 std::fmaf(x, y, z)。
  • CUDA C++ 标准库函数 cuda::std::fma(x, y, z) 和 cuda::std::fmaf(x, y, z)。

5.5.1.6. 点积示例(Dot Product Example)

考虑求两个短向量 (\overrightarrow{a}) 和 (\overrightarrow{b}) 的点积问题,两个向量均有四个元素。

[\begin{split}\overrightarrow{a} = \begin{bmatrix} a_{1} \ a_{2} \ a_{3} \ a_{4} \end{bmatrix} \qquad \overrightarrow{b} = \begin{bmatrix} b_{1} \ b_{2} \ b_{3} \ b_{4} \end{bmatrix} \qquad \overrightarrow{a} \cdot \overrightarrow{b} = a_{1}b_{1} + a_{2}b_{2} + a_{3}b_{3} + a_{4}b_{4}\end{split}]

尽管这个操作在数学上很容易写出来,但在软件中实现它涉及几种可能产生略微不同结果的替代方案。这里介绍的所有策略都使用完全符合 IEEE-754 标准的运算。

示例算法 1: 计算点积最简单的方法是使用顺序的乘积和,将乘法和加法分开进行。

最终结果可以表示为 (((((a_1 \times b_1) + (a_2 \times b_2)) + (a_3 \times b_3)) + (a_4 \times b_4))) 。

示例算法 2: 使用融合乘加顺序计算点积。

最终结果可以表示为 ((a_4 \times b_4) + ((a_3 \times b_3) + ((a_2 \times b_2) + (a_1 \times b_1 + 0)))) 。

示例算法 3: 使用分治策略计算点积。首先,我们计算向量前半部分和后半部分的点积。然后,我们使用加法合并这些结果。该算法被称为“并行算法”,因为两个子问题可以并行计算,因为它们彼此独立。然而,该算法并不要求并行实现;它可以用单个线程实现。

最终结果可以表示为 (((a_1 \times b_1) + (a_2 \times b_2)) + ((a_3 \times b_3) + (a_4 \times b_4))) 。

5.5.1.7. 舍入(Rounding)

IEEE-754 标准要求支持多种运算。这些运算包括算术运算,如加法、减法、乘法、除法、平方根、融合乘加、求余数、转换、缩放、符号和比较运算。对于给定的格式和舍入模式,这些运算的结果保证在所有标准实现中保持一致。


舍入模式

IEEE-754 标准定义了四种舍入模式:舍入到最近值、向正无穷舍入、向负无穷舍入 和 向零舍入。CUDA 支持所有四种模式。默认情况下,运算使用舍入到最近值。可以使用内部数学函数为单个运算选择其他舍入模式。

舍入模式解释
rn舍入到最近值,平局时取偶数
rz向零舍入
ru向 (\infty) 舍入
rd向 (-\infty) 舍入

5.5.1.8. 关于主机/设备计算精度的说明(Notes on Host/Device Computation Accuracy)

浮点计算结果的精度受多种因素影响。本节总结了在浮点计算中获得可靠结果的重要注意事项。其中一些方面在前面的章节中已有更详细的描述。 在比较 CPU 与 GPU 的结果时,这些方面也很重要。必须仔细解释主机与设备执行之间的差异。存在差异并不一定意味着 GPU 的结果不正确或 GPU 存在问题。

结合律:

有限精度下的浮点数加法和乘法不具有结合律,因为它们通常会产生无法在目标格式中直接表示的数学值,需要进行舍入。这些运算的求值顺序会影响舍入误差的累积方式,并可能显著改变最终结果。

融合乘加:

融合乘加运算在单次操作中计算 (a \times b + c),从而获得更高的精度和更快的执行时间。其使用会影响最终结果的精度。融合乘加依赖于硬件支持,可以通过显式调用相关函数或通过编译器优化标志隐式启用。

精度:

提高浮点精度有可能改善结果的准确性。更高的精度可以减少有效数字的损失,并能够表示更广泛的值。然而,更高精度的类型吞吐量较低,并且会消耗更多的寄存器。此外,使用它们来显式存储输入和输出会增加内存使用量和数据移动量。

编译器标志与优化:

所有主流编译器都提供多种优化标志来控制浮点运算的行为。GCC(-O3)、Clang(-O3)、nvcc(-O3)和 Microsoft Visual Studio(/O2)的最高优化级别不会影响浮点语义。然而,内联、循环展开、向量化和公共子表达式消除可能会影响结果。NVC++ 编译器也需要标志 -Kieee 和 -Mnofma 来获得符合 IEEE-754 标准的语义。有关影响浮点行为的选项的详细信息,请参阅 GCC、Clang、Microsoft Visual Studio Compiler、nvc++ 和 Arm C/C++ 编译器的文档。另请参阅 nvcc 用户手册,了解专门影响 CUDA 设备代码中浮点行为的编译器标志的详细说明:-ftz、-prec-div、-prec-sqrt、-fmad、--use_fast_math。除了这些浮点选项外,在用户程序的上下文中验证其他编译器优化的效果也很重要。鼓励用户通过广泛的测试来验证其结果的正确性,并比较启用优化与禁用所有设备代码优化时获得的结果;另请参阅 -G 编译器标志。

库实现:

IEEE-754 标准之外定义的函数不保证正确舍入,并且依赖于实现定义的行为。因此,结果可能在不同的平台之间存在差异,包括主机、设备以及不同的设备架构之间。 确定性结果:

确定性结果指的是在相同指定条件下,每次使用相同输入运行时,都能计算出相同的逐位数值输出。此类条件包括:

  • 硬件依赖性,例如在相同的 CPU 处理器或 GPU 设备上执行。
  • 编译器方面,例如编译器的版本以及
  • 编译器标志和优化
  • 影响计算的运行时条件,例如
  • 舍入模式
  • 或环境变量。
  • 计算的输入完全相同。
  • 线程配置,包括参与计算的线程数量及其组织方式,例如线程块和线程网格的大小。
  • 算术原子操作
  • 的顺序取决于硬件调度,这在各次运行之间可能有所不同。

利用 CUDA 库的优势:

CUDA 数学库 、 C 标准库数学函数 以及 C++ 标准库数学函数 旨在提升开发者在常见功能(尤其是浮点数学和数值密集型例程)上的生产力。这些功能提供了一致的高级接口,经过优化,并在各种平台和边界情况下经过了广泛测试。鼓励用户充分利用这些库,避免繁琐的手动重新实现。

5.5.2. 浮点数据类型(Floating-Point Data Types)

CUDA 支持 Bfloat16、半精度、单精度、双精度和四精度浮点数据类型。下表总结了 CUDA 支持的浮点数据类型及其要求。

精度 / 名称数据类型IEEE-754头文件 / 内置类型要求
Bfloat16__nv_bfloat16❌<cuda_bf16.h>计算能力 8.0 或更高。
半精度__half✔<cuda_fp16.h>
单精度float✔内置
双精度double✔内置
四精度__float128 / _Float128✔数学函数使用内置 <crt/device_fp128_functions.h>主机编译器支持且计算能力 10.0 或更高。C 或 C++ 的拼写(分别为 _Float128 和 __float128)也取决于主机编译器的支持。

CUDA 还支持 TensorFloat-32 (TF32)、微缩放 (MX) 浮点类型以及其他低精度数值格式,这些格式并非用于通用计算,而是用于涉及张量核心的专门用途。其中包括 4 位、6 位和 8 位浮点类型。更多详情请参阅 CUDA 数学 API。

下图展示了支持的浮点数据类型的尾数和指数大小。

浮点类型:尾数和指数大小 下表列出了支持的浮点数据类型的取值范围。

精度 / 名称最大值最小正值最小正非规格化数Epsilon
Bfloat16(\approx 2^{128})(\approx 3.39 \cdot 10^{38})(2^{-126})(\approx 1.18 \cdot 10^{-38})(2^{-133})(2^{-7})
半精度(\approx 2^{16})(65504)(2^{-14})(\approx 6.1 \cdot 10^{-5})(2^{-24})(2^{-10})
单精度(\approx 2^{128})(\approx 3.40 \cdot 10^{38})(2^{-126})(\approx 1.18 \cdot 10^{-38})(2^{-149})(2^{-23})
双精度(\approx 2^{1024})(\approx 1.8 \cdot 10^{308})(2^{-1022})(\approx 2.22 \cdot 10^{-308})(2^{-1074})(2^{-52})
四精度(\approx 2^{16384})(\approx 1.19 \cdot 10^{4932})(2^{-16382})(\approx 3.36 \cdot 10^{-4932})(2^{-16494})(2^{-112})

!!! note "提示" CUDA C++ 标准库在 <cuda/std/limits> 头文件中提供了 cuda::std::numeric_limits,用于查询支持的浮点类型(包括微缩放格式 MX)的属性和取值范围。有关可查询属性的列表,请参阅 C++ 参考。

复数支持:

  • CUDA C++ 标准库通过 <cuda/std/complex> 头文件中的 cuda::std::complex 类型支持复数。更多详细信息,请参阅 libcu++ 文档。
  • CUDA 还通过 cuComplex.h 头文件中的 cuComplex 和 cuDoubleComplex 类型提供对复数的基本支持。

5.5.3. CUDA 与 IEEE-754 合规性(CUDA and IEEE-754 Compliance)

所有 GPU 设备均遵循 IEEE 754-2019 二进制浮点运算标准,但存在以下限制:

  • 没有动态可配置的舍入模式;但是,大多数运算支持多个恒定的 IEEE 舍入模式,可通过特定命名的设备内部函数进行选择。
  • 没有检测浮点异常的机制,因此所有运算的行为都如同 IEEE-754 异常始终被屏蔽。如果发生异常事件,则传递 IEEE-754 定义的默认屏蔽响应。因此,虽然支持信令 NaN SNaN 编码,但它们不会发出信号,而是作为静默异常处理。
  • 浮点运算可能会改变输入 NaN 有效载荷的位模式。绝对值、取负等运算也可能不符合 IEEE 754 要求,这可能导致 NaN 的符号以实现定义的方式被更新。

为了最大限度地提高结果的可移植性,建议用户使用 nvcc 编译器浮点选项的默认设置:-ftz=false、-prec-div=true 和 -prec-sqrt=true,并且不要使用 --use_fast_math 选项。请注意,默认情况下允许浮点表达式重新关联和收缩,类似于 --fmad=true 选项。有关这些编译标志的详细说明,请参阅 nvcc 用户手册。 IEEE-754 和 C/C++ 语言标准并未明确规定当舍入后的整数值超出目标整数格式范围时,浮点值向整数值的转换行为。GPU 设备上的钳位行为在 PTX ISA 转换指令 章节中有详细说明。然而,当超出范围的转换不是直接通过 PTX 指令调用时,编译器优化可能会利用未定义行为条款,从而导致未定义行为并产生无效的 CUDA 程序。CUDA 数学文档在每个函数/内部函数的基础上向用户发出警告。例如,考虑 __double2int_rz() 指令。这可能与主机编译器和库实现的行为方式不同。

原子函数的非规格化数行为:

无论编译器标志 -ftz 的设置如何,原子操作在浮点非规格化数方面具有以下行为:

  • 对全局内存的原子单精度浮点加法始终以刷新到零模式运行,即行为等同于 PTX add.rn.ftz.f32 语义。
  • 对共享内存的原子单精度浮点加法始终支持非规格化数,即行为等同于 PTX add.rn.f32 语义。

5.5.4. CUDA 与 C/C++ 合规性(CUDA and C/C++ Compliance)

浮点异常:

与主机实现不同,设备代码支持的数学运算符和函数不会设置全局 errno 变量,也不会报告浮点异常来指示错误。因此,如果需要错误诊断机制,用户应为这些函数实现额外的输入和输出检查。

浮点操作的未定义行为:

数学运算常见的未定义行为条件包括:

  • 数学运算符和函数的无效参数:使用未初始化的浮点变量。在浮点变量的生命周期之外使用它。有符号整数溢出。解引用无效指针。
  • 浮点特有的未定义行为:将浮点值转换为结果无法表示的整数类型是未定义行为。这也包括 NaN 和无穷大。

用户有责任确保 CUDA 程序的有效性。无效参数可能导致未定义行为,并受编译器优化的影响。

与整数除以零不同,浮点数除以零不是未定义行为,也不受编译器优化的影响;相反,它是实现定义的行为。符合 IEC-60559 (IEEE-754) 的 C++ 实现(包括 CUDA)会产生无穷大。请注意,无效的浮点运算会产生 NaN,不应被误解为未定义行为。例如零除以零和无穷大除以无穷大。 浮点数字面量的可移植性:

C 和 C++ 都允许以十进制或十六进制表示法表示浮点值。十六进制浮点数字面量在 C99 和 C++17 中受支持,它表示一个可以用二进制精确表示的科学记数法实数。然而,这并不能保证该字面量会映射到目标变量中存储的实际值(参见下一段)。相反,十进制浮点数字面量可能表示一个无法用二进制表示的数值。

根据 C++ 标准规则,十六进制和十进制浮点数字面量会被舍入到最接近的可表示值(更大或更小),具体选择方式由实现定义。这种舍入行为在主机和设备之间可能不同。

float f1 = 0.5f;    // 0.5, '0.5f' 是十进制浮点数字面量
float f2 = 0x1p-1f; // 0.5, '0x1p-1f' 是十六进制浮点数字面量
float f3 = 0.1f;
// f1, f2 表示为 0 01111110 00000000000000000000000
// f3     表示为 0 01111011 10011001100110011001101

同一浮点表达式的运行时和编译时求值存在以下可移植性问题:

  • 浮点表达式的运行时求值可能受所选舍入模式、浮点收缩(FMA)和重关联编译器设置以及浮点异常的影响。请注意,CUDA 不支持浮点异常,且默认情况下舍入模式设置为“就近舍入,中间值向偶数舍入”。可以使用内部函数选择其他舍入模式。
  • 编译器可能对常量表达式使用更高精度的内部表示。
  • 编译器可能执行优化,例如常量折叠、常量传播和公共子表达式消除,这可能导致不同的最终值或比较结果。

C 标准数学库注意事项:

常见数学函数的主机实现以平台特定的方式映射到 C 标准数学库函数。这些函数由主机编译器提供,如果可用,则由相应的主机 libm 提供。

  • 主机编译器不提供的函数在 crt/math_functions.h 头文件中实现。例如,erfinv() 就在那里实现。
  • 不太常见的函数,如 rhypot() 和 cyl_bessel_i0(),仅在设备代码中可用。

如前所述,数学函数的主机和设备实现是独立的。有关这些函数行为的更多详细信息,请参阅主机实现的文档。


5.5.5. 浮点功能暴露(Floating-Point Functionality Exposure)

CUDA 支持的数学函数通过以下方式暴露: 内置 C/C++ 语言算术运算符:

  • x + y , x - y , x * y , x / y , x++ , x-- , x += y , x -= y , x *= y , x /= y 。
  • 支持单精度、双精度和四精度类型,分别为 float、double 和 __float128/_Float128。通过分别包含 <cuda_fp16.h> 和 <cuda_bf16.h> 头文件,也支持 __half 和 __nv_bfloat16 类型。__float128/_Float128 类型的支持依赖于主机编译器和设备计算能力,请参阅支持的浮点类型表。
  • 它们在主机和设备代码中均可用。
  • 它们的行为受 nvcc 优化标志影响。

CUDA C++ 标准库数学函数:

  • 通过 <cuda/std/cmath> 头文件和 cuda::std:: 命名空间公开完整的 C++ <cmath> 头文件函数集。
  • 支持 IEEE-754 标准浮点类型 __half、float、double、__float128,以及 Bfloat16 __nv_bfloat16。__float128 支持依赖于主机编译器和设备计算能力,请参阅支持的浮点类型表。
  • 它们在主机和设备代码中均可用。
  • 它们通常依赖于 CUDA 数学 API 函数。因此,主机和设备代码之间可能存在不同的精度级别。
  • 它们的行为受 nvcc 优化标志影响。
  • 根据 C++23 和 C++26 标准规范,其功能子集也支持常量表达式,例如 constexpr 函数。

CUDA C 标准库数学函数 (CUDA 数学 API):

  • 公开 C <math.h> 头文件函数的一个子集。
  • 支持单精度和双精度类型,分别为 float 和 double。它们在主机和设备代码中均可用。它们不需要额外的头文件。它们的行为受 nvcc 优化标志影响。
  • <math.h> 头文件功能的一个子集也可用于 __half、__nv_bfloat16 和 __float128/_Float128 类型。这些函数的名称类似于 C 标准库中的函数。__half 和 __nv_bfloat16 类型分别需要 <cuda_fp16.h> 和 <cuda_bf16.h> 头文件。它们在主机和设备代码中的可用性是按函数定义的。__float128/_Float128 类型的支持依赖于主机编译器和设备计算能力,请参阅支持的浮点类型表。相关函数需要 crt/device_fp128_functions.h 头文件,并且仅在设备代码中可用。
  • 它们在主机和设备代码之间可能具有不同的精度。

非标准 CUDA 数学函数 (CUDA 数学 API):

  • 公开不属于 C/C++ 标准库的数学功能。
  • 主要支持单精度和双精度类型,分别为 float 和 double。它们在主机和设备代码中的可用性是按函数定义的。它们不需要额外的头文件。它们在主机和设备代码之间可能具有不同的精度。
  • __nv_bfloat16、__half、__float128/_Float128 仅支持有限的一组函数。__half 和 __nv_bfloat16 类型分别需要包含 <cuda_fp16.h> 和 <cuda_bf16.h> 头文件。__float128/_Float128 类型的支持依赖于主机编译器和设备计算能力,请参阅支持的浮点类型表。相关函数需要包含 crt/device_fp128_functions.h 头文件。它们仅在设备代码中可用。
  • 它们的行为受 nvcc 优化标志影响。

内建数学函数 (CUDA 数学 API):

  • 支持单精度和双精度类型,分别是 float 和 double。
  • 它们仅在设备代码中可用。
  • 它们比相应的 CUDA 数学 API 函数更快,但精度较低。
  • 它们的行为不受 nvcc 浮点优化标志 -prec-div=false、-prec-sqrt=false 和 -fmad=true 的影响。唯一的例外是 -ftz=true,它也包含在 --use_fast_math 中。
功能支持的类型主机设备受浮点优化标志影响(仅针对 float 和 double)
内建 C/C++ 语言算术运算符float, double, __half, __nv_bfloat16, __float128/_Float128, cuda::std::complex✔✔✔
CUDA C++ 标准库数学函数float, double, __half, __nv_bfloat16, __float128, cuda::std::complex✔✔✔
__nv_fp8_e4m3, __nv_fp8_e5m2, __nv_fp8_e8m0, __nv_fp6_e2m3, __nv_fp6_e3m2, __nv_fp4_e2m1 *
CUDA C 标准库数学函数float, double✔✔✔
__nv_bfloat16, __half 有限支持及类似名称基于逐个函数
__float128/_Float128 有限支持及类似名称✘✔
非标准 CUDA 数学函数float, double基于逐个函数✔
__nv_bfloat16, __half, __float128/_Float128 有限支持✘✔
内建函数float, double✘✔仅受 -ftz=true 影响(也包含在 --use_fast_math 中)

***** CUDA C++ 标准库函数 支持对小浮点类型进行查询,例如 numeric_limits<T>、fpclassify()、isfinite()、isnormal()、isinf() 和 isnan()。

以下部分在适用时提供了其中一些函数的精度信息。它使用 ULP 进行量化。有关最后一位单位 (ULP) 定义的更多信息,请参阅 Jean-Michel Muller 的论文 On the definition of ulp(x)。

5.5.6. 内置算术运算符(Built-In Arithmetic Operators)

对于单精度、双精度和四精度类型,内置的 C/C++ 语言运算符,例如 x + y、x - y、x * y、x / y、x++、x-- 以及倒数 1 / x,均符合 IEEE-754 标准。在使用向偶数舍入的舍入模式下,它们保证最大 ULP 误差为零。这些运算符在主机和设备代码中均可用。

nvcc 编译标志 -fmad=true(也包含在 --use_fast_math 中)允许将浮点乘法和加法/减法合并为浮点乘加运算,这对单精度类型 float 的最大 ULP 误差有以下影响:

  • x * y + z → __fmaf_rn(x, y, z) : 0 ULP

nvcc 编译标志 -prec-div=false(也包含在 --use_fast_math 中)对单精度类型 float 的除法运算符 / 的最大 ULP 误差有以下影响:

  • x / y → __fdividef(x, y) : 2 ULP
  • 1 / x : 1 ULP

5.5.7. CUDA C++ 数学标准库函数(CUDA C++ Mathematical Standard Library Functions)

CUDA 通过 cuda::std:: 命名空间为 C++ 标准库数学函数 提供全面支持。这些功能是 <cuda/std/cmath> 头文件的一部分。它们在主机和设备代码中均可用。

以下部分指定了与 CUDA 数学 API 的映射关系,以及每个函数在设备上执行时的误差界限。

  • 最大 ULP 误差被表述为:函数返回值与根据向偶数舍入模式获得的相应精度的正确舍入结果之间,以 ULP 为单位的差值绝对值的最大观测值。
  • 误差界限来源于广泛(尽管不是穷尽)的测试。因此,它们不能得到保证。

5.5.7.1. 基本运算(Basic Operations)

CUDA 数学 API 中的基本运算函数在主机和设备代码中均可用,__float128 除外。

以下所有函数的最大 ULP 误差均为零。

cuda::std 函数含义__nv_bfloat16__halffloatdouble__float128
fabs(x)(x)__habs(x)__habs(x)fabsf(x)
fmod(x, y)(\dfrac{x}{y}) 的余数,计算为 (x - \mathrm{trunc}\left(\dfrac{x}{y}\right) \cdot y)N/AN/Afmodf(x, y)fmod(x, y)__nv_fp128_fmod(x, y)
remainder(x, y)(\dfrac{x}{y}) 的余数,计算为 (x - \mathrm{rint}\left(\dfrac{x}{y}\right) \cdot y)N/AN/Aremainderf(x, y)remainder(x, y)__nv_fp128_remainder(x, y)
remquo(x, y, iptr)(\dfrac{x}{y}) 的余数和商N/AN/Aremquof(x, y, iptr)remquo(x, y, iptr)N/A
fma(x, y, z)(x \cdot y + z)__hfma(x, y, z) , 仅设备__hfma(x, y, z) , 仅设备fmaf(x, y, z)fma(x, y, z)__nv_fp128_fma(x, y, z)
cuda::std 函数含义__nv_bfloat16__halffloatdouble__float128
---------------------
fmax(x, y)(\max(x, y))__hmax(x, y)__hmax(x, y)fmaxf(x, y)fmax(x, y)__nv_fp128_fmax(x, y)
fmin(x, y)(\min(x, y))__hmin(x, y)__hmin(x, y)fminf(x, y)fmin(x, y)__nv_fp128_fmin(x, y)
fdim(x, y)(\max(x-y, 0))N/AN/Afdimf(x, y)fdim(x, y)__nv_fp128_fdim(x, y)
nan(str)从字符串表示生成 NaN 值N/AN/Ananf(str)nan(str)N/A

*****标记为“N/A”的数学函数对于 CUDA 扩展浮点类型(例如 __half 和 __nv_bfloat16)并非原生可用。在这些情况下,函数通过转换为 float 类型,然后将结果转换回来进行模拟。

5.5.7.2. 指数函数(Exponential Functions)

CUDA 数学 API 中的指数函数在主机和设备代码中仅对 float 和 double 类型可用。

cuda::std 函数含义__nv_bfloat16__halffloatdouble__float128
exp(x)(e^x)hexp(x) 0 ULPhexp(x) 0 ULPexpf(x) 2 ULPexp(x) 1 ULP__nv_fp128_exp(x) 1 ULP
exp2(x)(2^x)hexp2(x) 0 ULPhexp2(x) 0 ULPexp2f(x) 2 ULPexp2(x) 1 ULP__nv_fp128_exp2(x) 1 ULP
expm1(x)(e^x - 1)N/AN/Aexpm1f(x) 1 ULPexpm1(x) 1 ULP__nv_fp128_expm1(x) 1 ULP
log(x)(\ln(x))hlog(x) 0 ULPhlog(x) 0 ULPlogf(x) 1 ULPlog(x) 1 ULP__nv_fp128_log(x) 1 ULP
log10(x)(\log_{10}(x))hlog10(x) 0 ULPhlog10(x) 0 ULPlog10f(x) 2 ULPlog10(x) 1 ULP__nv_fp128_log10(x) 1 ULP
log2(x)(\log_2(x))hlog2(x) 0 ULPhlog2(x) 0 ULPlog2f(x) 1 ULPlog2(x) 1 ULP__nv_fp128_log2(x) 1 ULP
log1p(x)(\ln(1+x))N/AN/Alog1pf(x) 1 ULPlog1p(x) 1 ULP__nv_fp128_log1p(x) 1 ULP

*****标记为“N/A”的数学函数对于 CUDA 扩展浮点类型(例如 __half 和 __nv_bfloat16)并非原生可用。在这些情况下,函数通过转换为 float 类型,然后将结果转换回来进行模拟。

5.5.7.3. 幂函数(Power Functions)

CUDA 数学 API 中的幂函数在主机和设备代码中仅对 float 和 double 类型可用。

cuda::std 函数含义__nv_bfloat16__halffloatdouble__float128
pow(x, y)(x^y)N/AN/Apowf(x, y) 4 ULPpow(x, y) 2 ULP__nv_fp128_pow(x, y) 1 ULP
sqrt(x)(\sqrt{x})hsqrt(x) 0 ULPhsqrt(x) 0 ULPsqrtf(x) • 0 ULP • 1 ULP with --use_fast_mathsqrt(x) 0 ULP__nv_fp128_sqrt(x) 0 ULP
cbrt(x)(\sqrt[3]{x})N/AN/Acbrtf(x) 1 ULPcbrt(x) 1 ULPN/A
hypot(x, y)(\sqrt{x^2 + y^2})N/AN/Ahypotf(x, y) 3 ULPhypot(x, y) 2 ULP__nv_fp128_hypot(x, y) 1 ULP

*****标记为“N/A”的数学函数对于 CUDA 扩展浮点类型(例如 __half 和 __nv_bfloat16)并非原生可用。在这些情况下,函数通过转换为 float 类型,然后将结果转换回来进行模拟。

5.5.7.4. 三角函数(Trigonometric Functions)

CUDA Math API 中的三角函数在主机和设备代码中均可用,但仅适用于 float 和 double 类型。

cuda::std 函数含义__nv_bfloat16__halffloatdouble__float128
sin(x)(\sin(x))hsin(x) 0 ULPhsin(x) 0 ULPsinf(x) 2 ULPsin(x) 2 ULP__nv_fp128_sin(x) 1 ULP
cos(x)(\cos(x))hcos(x) 0 ULPhcos(x) 0 ULPcosf(x) 2 ULPcos(x) 2 ULP__nv_fp128_cos(x) 1 ULP
tan(x)(\tan(x))N/AN/Atanf(x) 4 ULPtan(x) 2 ULP__nv_fp128_tan(x) 1 ULP
asin(x)(\sin^{-1}(x))N/AN/Aasinf(x) 2 ULPasin(x) 2 ULP__nv_fp128_asin(x) 1 ULP
acos(x)(\cos^{-1}(x))N/AN/Aacosf(x) 2 ULPacos(x) 2 ULP__nv_fp128_acos(x) 1 ULP
atan(x)(\tan^{-1}(x))N/AN/Aatanf(x) 2 ULPatan(x) 2 ULP__nv_fp128_atan(x) 1 ULP
atan2(y, x)(\tan^{-1}\left(\dfrac{y}{x}\right))N/AN/Aatan2f(y, x) 3 ULPatan2(y, x) 2 ULPN/A

*****标记为“N/A”的数学函数对于 CUDA 扩展浮点类型(例如 __half 和 __nv_bfloat16)并非原生可用。在这些情况下,函数通过转换为 float 类型然后转换回结果来模拟。

5.5.7.5. 双曲函数(Hyperbolic Functions)

CUDA Math API 中的双曲函数在主机和设备代码中均可用,但仅适用于 float 和 double 类型。

cuda::std 函数含义__nv_bfloat16__halffloatdouble__float128
sinh(x)(\sinh(x))N/AN/Asinhf(x) 3 ULPsinh(x) 2 ULP__nv_fp128_sinh(x) 1 ULP
cosh(x)(\cosh(x))N/AN/Acoshf(x) 2 ULPcosh(x) 1 ULP__nv_fp128_cosh(x) 1 ULP
tanh(x)(\tanh(x))htanh(x) 0 ULPhtanh(x) 0 ULPtanhf(x) 2 ULPtanh(x) 1 ULP__nv_fp128_tanh(x) 1 ULP
asinh(x)(\operatorname{sinh}^{-1}(x))N/AN/Aasinhf(x) 3 ULPasinh(x) 3 ULP__nv_fp128_asinh(x) 1 ULP
acosh(x)(\operatorname{cosh}^{-1}(x))N/AN/Aacoshf(x) 4 ULPacosh(x) 3 ULP__nv_fp128_acosh(x) 1 ULP
atanh(x)(\operatorname{tanh}^{-1}(x))N/AN/Aatanhf(x) 3 ULPatanh(x) 2 ULP__nv_fp128_atanh(x) 1 ULP

*****标记为“N/A”的数学函数对于 CUDA 扩展浮点类型(例如 __half 和 __nv_bfloat16)并非原生可用。在这些情况下,函数通过转换为 float 类型然后转换回结果来模拟。

5.5.7.6. 误差函数和 Gamma 函数(Error and Gamma Functions)

CUDA Math API 中的误差函数和 Gamma 函数在主机和设备代码中均可用,适用于 float 和 double 类型。

误差函数和 Gamma 函数对于 CUDA 扩展浮点类型(例如 __half 和 __nv_bfloat16)并非原生可用。在这些情况下,函数通过转换为 float 类型然后转换回结果来模拟。

cuda::std 函数含义floatdouble
erf(x)(\dfrac{2}{\sqrt{\pi}} \int_0^x e^{-t^2} dt)erff(x) 2 ULPerf(x) 2 ULP
erfc(x)(1 - \mathrm{erf}(x))erfcf(x) 4 ULPerfc(x) 5 ULP
tgamma(x)(\Gamma(x))tgammaf(x) 5 ULPtgamma(x) 10 ULP
lgamma(x)(\ln\Gamma(x))

5.5.7.7. 最近整数浮点运算(Nearest Integer Floating-Point Operations)

CUDA Math API 中关于最近整数浮点运算的函数在主机和设备代码中均可用,但仅适用于 float 和 double 类型。

以下所有函数的最大 ULP 误差为零。

cuda::std 函数含义__nv_bfloat16__halffloatdouble__float128
ceil(x)(\lceil x \rceil)hceil(x)hceil(x)ceilf(x)ceil(x)__nv_fp128_ceil(x)
floor(x)(\lfloor x \rfloor)hfloor(x)hfloor(x)floorf(x)floor(x)__nv_fp128_floor(x)
trunc(x)截断为整数htrunc(x)htrunc(x)truncf(x)trunc(x)__nv_fp128_trunc(x)
round(x)四舍五入到最近的整数,中间值远离零N/AN/Aroundf(x)round(x)__nv_fp128_round(x)
nearbyint(x)四舍五入到最近的整数,中间值取偶N/AN/Anearbyintf(x)nearbyint(x)N/A
rint(x)四舍五入到最近的整数,中间值取偶hrint(x)hrint(x)rintf(x)rint(x)__nv_fp128_rint(x)
lrint(x)四舍五入到最近的整数,中间值取偶(返回 long int)N/AN/Alrintf(x)lrint(x)N/A
llrint(x)四舍五入到最近的整数,中间值取偶(返回 long long int)N/AN/Allrintf(x)llrint(x)N/A
lround(x)四舍五入到最近的整数,中间值远离零(返回 long int)N/AN/Alroundf(x)lround(x)N/A
llround(x)四舍五入到最近的整数,中间值远离零(返回 long long int)N/AN/Allroundf(x)llround(x)N/A

*****标记为“N/A”的数学函数对于 CUDA 扩展浮点类型(如 __half 和 __nv_bfloat16)并非原生可用。在这些情况下,函数通过转换为 float 类型然后转换回结果来模拟。

性能考虑

将单精度或双精度浮点操作数舍入为整数的推荐方法是使用函数 rintf() 和 rint(),而不是 roundf() 和 round()。这是因为 roundf() 和 round() 在设备代码中映射到多条指令,而 rintf() 和 rint() 映射到单条指令。truncf()、trunc()、ceilf()、ceil()、floorf() 和 floor() 也各自映射到单条指令。

5.5.7.8. 浮点操作函数(Floating-Point Manipulation Functions)

CUDA Math API 中关于浮点操作函数的函数在主机和设备代码中均可用,但 __float128 除外。 浮点数操作函数并非原生支持 CUDA 扩展浮点类型,例如 __half 和 __nv_bfloat16。在这些情况下,函数通过转换为 float 类型,然后将结果转换回原类型来模拟实现。

以下所有函数均具有零最大 ULP 误差。

cuda::std 函数含义floatdouble__float128
frexp(x, exp)提取尾数和指数frexpf(x, exp)frexp(x, exp)__nv_fp128_frexp(x, nptr)
ldexp(x, n)(x \cdot 2^{\mathrm{n}})ldexpf(x, n)ldexp(x, n)__nv_fp128_ldexp(x, n)
modf(x, iptr)提取整数和小数部分modff(x, iptr)modf(x, iptr)__nv_fp128_modf(x, iptr)
scalbn(x, n)(x \cdot 2^n)scalbnf(x, n)scalbn(x, n)N/A
scalbln(x, n)(x \cdot 2^n)scalblnf(x, n)scalbln(x, n)N/A
ilogb(x)(\lfloor \log_2(x) \rfloor)ilogbf(x)
logb(x)(\lfloor \log_2(x) \rfloor)logbf(x)
nextafter(x, y)朝向 (y) 的下一个可表示值nextafterf(x, y)nextafter(x, y)N/A
copysign(x, y)将 (y) 的符号复制到 (x)copysignf(x, y)copysign(x, y)__nv_fp128_copysign(x, y)

5.5.7.9. 分类与比较(Classification and Comparison)

CUDA 数学 API 中的分类和比较函数在主机和设备代码中均可用,但 __float128 除外。

以下所有函数均具有零最大 ULP 误差。

cuda::std 函数含义__nv_bfloat16__halffloatdouble__float128
fpclassify(x)对 (x) 进行分类N/AN/AN/AN/AN/A
isfinite(x)检查 (x) 是否为有限值N/AN/Aisfinite(x)isfinite(x)N/A
isinf(x)检查 (x) 是否为无穷大__hisinf(x)__hisinf(x)isinf(x)isinf(x)N/A
isnan(x)检查 (x) 是否为 NaN__hisnan(x)__hisnan(x)isnan(x)isnan(x)__nv_fp128_isnan(x)
isnormal(x)检查 (x) 是否为规格化数N/AN/AN/AN/AN/A
signbit(x)检查符号位是否被设置N/AN/Asignbit(x)signbit(x)N/A
isgreater(x, y)检查 (x > y)__hgt(x, y)__hgt(x, y)N/AN/AN/A
isgreaterequal(x, y)检查 (x \geq y)__hge(x, y)__hge(x, y)N/AN/AN/A
isless(x, y)检查 x < y__hlt(x, y)__hlt(x, y)N/AN/AN/A
islessequal(x, y)检查 (x \leq y)__hle(x, y)__hle(x, y)N/AN/AN/A
islessgreater(x, y)检查 x < y 或 x > y__hne(x, y)__hne(x, y)N/AN/AN/A
isunordered(x, y)检查 (x)、(y) 或两者是否为 NaNN/AN/AN/AN/A__nv_fp128_isunordered(x, y)

*****标记为“N/A”的数学函数并非原生支持 CUDA 扩展浮点类型,例如 __half 和 __nv_bfloat16。

5.5.8. 非标准 CUDA 数学函数(Non-Standard CUDA Mathematical Functions)

CUDA 提供了不属于 C/C++ 标准库的数学函数,它们作为扩展提供。对于单精度和双精度函数,主机和设备代码的可用性是按每个函数单独定义的。

本节规定了每个函数在设备上执行时的误差界限。

  • 最大 ULP 误差表示为:根据“向最接近偶数舍入”模式获得的相应精度的正确舍入结果,与函数返回值之间的 ULP 差值绝对值的最大观测值。
  • 误差界限来源于广泛但非穷尽的测试。因此,不能保证其绝对性。
含义floatdouble
(\dfrac{x}{y})fdividef(x, y) , 仅设备 0 ULP,与 x / y 相同N/A
(10^x)exp10f(x) 2 ULPexp10(x) 1 ULP
(\sqrt{x^2 + y^2 + z^2})norm3df(x, y, z) , 仅设备 3 ULPnorm3d(x, y, z) , 仅设备 2 ULP
(\sqrt{x^2 + y^2 + z^2 + t^2})norm4df(x, y, z, t) , 仅设备 3 ULPnorm4d(x, y, z, t) , 仅设备 2 ULP
(\sqrt{\sum_{i=0}^{\mathrm{dim}-1} p_i^{2}})normf(dim, p) , 仅设备 由于使用了快速算法,会因舍入导致精度损失,无法提供误差界限norm(dim, p) , 仅设备 由于使用了快速算法,会因舍入导致精度损失,无法提供误差界限
(\dfrac{1}{\sqrt{x}})rsqrtf(x) 2 ULPrsqrt(x) 1 ULP
(\dfrac{1}{\sqrt[3]{x}})rcbrtf(x) 1 ULPrcbrt(x) 1 ULP
(\dfrac{1}{\sqrt{x^2 + y^2}})rhypotf(x, y) , 仅设备 2 ULPrhypot(x, y) , 仅设备 1 ULP
(\dfrac{1}{\sqrt{x^2 + y^2 + z^2}})rnorm3df(x, y, z) , 仅设备 2 ULPrnorm3d(x, y, z) , 仅设备 1 ULP
(\dfrac{1}{\sqrt{x^2 + y^2 + z^2 + t^2}})rnorm4df(x, y, z, t) , 仅设备 2 ULPrnorm4d(x, y, z, t) , 仅设备 1 ULP
(\dfrac{1}{\sqrt{\sum_{i=0}^{\mathrm{dim}-1} p_i^{2}}})rnormf(dim, p) , 仅设备 由于使用了快速算法,会因舍入导致精度损失,无法提供误差界限rnorm(dim, p) , 仅设备 由于使用了快速算法,会因舍入导致精度损失,无法提供误差界限
(\cos(\pi x))cospif(x) 1 ULPcospi(x) 2 ULP
(\sin(\pi x))sinpif(x) 1 ULPsinpi(x) 2 ULP
(\sin(\pi x), \cos(\pi x))sincospif(x, sptr, cptr) 1 ULPsincospi(x, sptr, cptr) 2 ULP
(\Phi(x))normcdff(x) 5 ULPnormcdf(x) 5 ULP
(\Phi^{-1}(x))normcdfinvf(x) 5 ULPnormcdfinv(x) 8 ULP
(\mathrm{erfc}^{-1}(x))erfcinvf(x) 4 ULPerfcinv(x) 6 ULP
(e^{x^2}\mathrm{erfc}(x))erfcxf(x) 4 ULPerfcx(x) 4 ULP
(\mathrm{erf}^{-1}(x))erfinvf(x) 2 ULPerfinv(x) 5 ULP
(I_0(x))cyl_bessel_i0f(x) , 仅设备 6 ULPcyl_bessel_i0(x) , 仅设备 6 ULP
(I_1(x))cyl_bessel_i1f(x) , 仅设备 6 ULPcyl_bessel_i1(x) , 仅设备 6 ULP
(J_0(x))j0f(x) • 对于 `x
(J_1(x))j1f(x) • 当 `x
(J_n(x))jnf(n, x) 对于 (n = 128),最大绝对误差 (= 2.2 \cdot 10^{-6})jn(n, x) 对于 (n = 128),最大绝对误差 (= 5 \cdot 10^{-12})
(Y_0(x))y0f(x) • 当 `x
(Y_1(x))y1f(x) • 当 `x
(Y_n(x))ynf(n, x) • 当 `x

适用于 __half、__nv_bfloat16 和 __float128/_Float128 的非标准 CUDA 数学函数仅在设备代码中可用。

含义__nv_bfloat16__half__float128/_Float128
(\dfrac{1}{x})hrcp(x) 0 ULPhrcp(x) 0 ULPN/A
(10^x)hexp10(x) 0 ULPhexp10(x) 0 ULP__nv_fp128_exp10(x) 1 ULP
(\dfrac{1}{\sqrt{x}})hrsqrt(x) 0 ULPhrsqrt(x) 0 ULPN/A
(\tanh(x)) (近似)htanh_approx(x) 1 ULPhtanh_approx(x) 1 ULPN/A

5.5.9. 内置函数(Intrinsic Functions)

内置数学函数是其对应的 CUDA C 标准库数学函数 的更快但精度较低的版本。

  • 它们具有相同的名称,但前缀为 __,例如 __sinf(x)。
  • 它们仅在设备代码中可用。
  • 它们更快,因为它们映射到更少的原生指令。
  • 标志 --use_fast_math 会自动将相应的 CUDA 数学 API 函数转换为内置函数。有关受影响函数的完整列表,请参阅 --use_fast_math 效果部分。

5.5.9.1. 基本内置函数(Basic Intrinsic Functions)

一部分数学内置函数允许指定舍入模式:

  • 后缀为 _rn 的函数使用“舍入到最接近的偶数”舍入模式。
  • 后缀为 _rz 的函数使用“向零舍入”舍入模式。
  • 后缀为 _ru 的函数使用“向上舍入”(向正无穷大)舍入模式。
  • 后缀为 _rd 的函数使用“向下舍入”(向负无穷大)舍入模式。

__fadd_[rn,rz,ru,rd]()、__dadd_[rn,rz,ru,rd]()、__fmul_[rn,rz,ru,rd]() 和 __dmul_[rn,rz,ru,rd]() 函数映射到加法和乘法运算,编译器永远不会将这些运算合并到 FFMA 或 DFMA 指令中。相比之下,由 * 和 + 运算符生成的加法和乘法经常被合并到 FFMA 或 DFMA 中。 下表列出了单精度和双精度浮点内置函数。所有这些函数都具有 0 的最大 ULP 误差,并且符合 IEEE 标准。

含义floatdouble
(x + y)__fadd_[rn,rz,ru,rd](x, y)__dadd_[rn,rz,ru,rd](x, y)
(x - y)__fsub_[rn,rz,ru,rd](x, y)__dsub_[rn,rz,ru,rd](x, y)
(x \cdot y)__fmul_[rn,rz,ru,rd](x, y)__dmul_[rn,rz,ru,rd](x, y)
(x \cdot y + z)__fmaf_[rn,rz,ru,rd](x, y, z)__fma_[rn,rz,ru,rd](x, y, z)
(\dfrac{x}{y})__fdiv_[rn,rz,ru,rd](x, y)__ddiv_[rn,rz,ru,rd](x, y)
(\dfrac{1}{x})__frcp_[rn,rz,ru,rd](x)__drcp_[rn,rz,ru,rd](x)
(\sqrt{x})__fsqrt_[rn,rz,ru,rd](x)__dsqrt_[rn,rz,ru,rd](x)

5.5.9.2. 仅限单精度的内置函数(Single-Precision-Only Intrinsic Functions)

下表列出了单精度浮点内置函数及其最大 ULP 误差。

  • 最大 ULP 误差定义为:函数返回值与根据“就近舍入,偶数优先”舍入模式得到的对应精度的正确舍入结果之间,以 ULP 为单位的差值绝对值的最大观测值。
  • 误差界限来源于广泛但非穷尽的测试。因此,不能保证绝对准确。
函数含义最大 ULP 误差
__fdividef(x, y)(\dfrac{x}{y})对于 (
__frsqrt_rn(x)(\dfrac{1}{\sqrt{x}})0 ULP
__expf(x)(e^x)(2 + \lfloor
__exp10f(x)(10^x)(2 + \lfloor
__powf(x, y)(x^y)由 exp2f(y * __log2f(x)) 推导得出
__logf(x)(\ln(x))• 对于 (x \in [0.5, 2]),绝对误差为 (2^{-21.41}) • 其他情况下为 3 ULP
__log2f(x)(\log_2(x))• 对于 (x \in [0.5, 2]),绝对误差为 (2^{-22}) • 其他情况下为 2 ULP
__log10f(x)(\log_{10}(x))• 对于 (x \in [0.5, 2]),绝对误差为 (2^{-24}) • 其他情况下为 3 ULP
__sinf(x)(\sin(x))• 对于 (x \in [-\pi, \pi]),绝对误差为 (2^{-21.41}) • 其他情况下更大
__cosf(x)(\cos(x))• 对于 (x \in [-\pi, \pi]),绝对误差为 (2^{-21.41}) • 其他情况下更大
__sincosf(x, sptr, cptr)(\sin(x), \cos(x))分量误差分别与 __sinf(x) 和 __cosf(x) 相同
__tanf(x)(\tan(x))由 __sinf(x) * (1 / __cosf(x)) 推导得出
__tanhf(x)(\tanh(x))• 最大相对误差:(2^{-11}) • 即使在 -ftz=true 编译器标志下,次正规结果也不会被刷新为零。

5.5.9.3. --use_fast_math 的影响(--use_fast_math Effect)

nvcc 编译器标志 --use_fast_math 会将设备代码中调用的 CUDA 数学 API 函数 的一个子集转换为其对应的内置函数。请注意,CUDA C++ 标准库函数 也会受此标志影响。关于使用内置函数代替 CUDA 数学 API 函数的影响,更多细节请参见 内置函数 部分。

一种更稳健的方法是,仅在性能提升具有足够价值,且精度降低和特殊情形处理方式改变等特性变化可接受的情况下,有选择地将数学函数调用替换为内联函数版本。

设备函数内联函数
x/y, fdividef(x, y)__fdividef(x, y)
sinf(x)__sinf(x)
cosf(x)__cosf(x)
tanf(x)__tanf(x)
sincosf(x, sptr, cptr)__sincosf(x, sptr, cptr)
logf(x)__logf(x)
log2f(x)__log2f(x)
log10f(x)__log10f(x)
expf(x)__expf(x)
exp10f(x)__exp10f(x)
powf(x,y)__powf(x,y)
tanhf(x)__tanhf(x)

5.5.10. 参考文献(References)

  1. IEEE 754-2019 浮点运算标准。
  2. Jean-Michel Muller. On the definition of ulp(x) . INRIA/LIP 研究报告,2005。
  3. Nathan Whitehead, Alex Fit-Florea. Precision & Performance: Floating Point and IEEE 754 Compliance for NVIDIA GPUs . Nvidia 技术报告,2011。
  4. David Goldberg. What every computer scientist should know about floating-point arithmetic . ACM 计算概览,1991年3月。
  5. David Monniaux. The pitfalls of verifying floating-point computations . ACM 编程语言与系统汇刊,2008年5月。
  6. Peter Dinda, Conor Hetland. Do Developers Understand IEEE Floating Point? . IEEE 国际并行与分布式处理研讨会 (IPDPS),2018。

本页


官方交叉引用与 API 链接(Official Cross-References)

下列链接来自官方源文件中未能在译文正文中就近展开的锚点和 API 引用;保留它们可避免翻译版丢失原文的脚注、标准和 API 交叉引用。

  • cuda::std::numeric_limits<T>::quiet_NaN
  • cuda::std::numeric_limits<T>::signaling_NaN
  • C Standard Library functions
  • CUDA mathematical intrinsic functions
  • CUDA C++ Standard Library functions
  • GCC
  • Clang
  • NVC++
  • Microsoft Visual Studio
  • associative
  • Fused Multiply-Add
  • GCC
  • Clang
  • Microsoft Visual Studio Compiler
  • Arm C/C++ compiler
  • Compiler Flags and Optimizations
  • rounding mode
  • arithmetic atomic operations
  • CUDA Math Libraries
  • C++ Standard Library Mathematical functions
  • cuda::std::complex
  • libcu++ documentation
  • 支持的浮点类型
  • 头文件函数
  • optimization flags
  • __fmaf_rn(x, y, z)
  • __fdividef(x, y)
  • fabs(x)
  • __habs(x)
  • __habs(x)
  • fabsf(x)
  • fabs(x)
  • __nv_fp128_fabs(x)
  • fmod(x, y)
  • fmodf(x, y)
  • fmod(x, y)
  • __nv_fp128_fmod(x, y)
  • remainder(x, y)
  • remainderf(x, y)
  • remainder(x, y)
  • __nv_fp128_remainder(x, y)
  • remquo(x, y, iptr)
  • remquof(x, y, iptr)
  • remquo(x, y, iptr)
  • __hfma(x, y, z)
  • __hfma(x, y, z)
  • fmaf(x, y, z)
  • fma(x, y, z)
  • __nv_fp128_fma(x, y, z)
  • fmax(x, y)
  • __hmax(x, y)
  • __hmax(x, y)
  • fmaxf(x, y)
  • fmax(x, y)
  • __nv_fp128_fmax(x, y)
  • fmin(x, y)
  • __hmin(x, y)
  • __hmin(x, y)
  • fminf(x, y)
  • fmin(x, y)
  • __nv_fp128_fmin(x, y)
  • fdim(x, y)
  • fdimf(x, y)
  • fdim(x, y)
  • __nv_fp128_fdim(x, y)
  • nan(str)
  • nanf(str)
  • nan(str)
  • exp(x)
  • hexp(x)
  • hexp(x)
  • expf(x)
  • exp(x)
  • __nv_fp128_exp(x)
  • exp2(x)
  • hexp2(x)
  • hexp2(x)
  • exp2f(x)
  • exp2(x)
  • __nv_fp128_exp2(x)
  • expm1(x)
  • expm1f(x)
  • expm1(x)
  • __nv_fp128_expm1(x)
  • log(x)
  • hlog(x)
  • hlog(x)
  • logf(x)
  • log(x)
  • __nv_fp128_log(x)
  • log10(x)
  • hlog10(x)
  • hlog10(x)
  • log10f(x)
  • log10(x)
  • __nv_fp128_log10(x)
  • log2(x)
  • hlog2(x)
  • hlog2(x)
  • log2f(x)
  • log2(x)
  • __nv_fp128_log2(x)
  • log1p(x)
  • log1pf(x)
  • log1p(x)
  • __nv_fp128_log1p(x)
  • pow(x, y)
  • powf(x, y)
  • pow(x, y)
  • __nv_fp128_pow(x, y)
  • sqrt(x)
  • hsqrt(x)
  • hsqrt(x)
  • sqrtf(x)
  • sqrt(x)
  • __nv_fp128_sqrt(x)
  • cbrt(x)
  • cbrtf(x)
  • cbrt(x)
  • hypot(x, y)
  • hypotf(x, y)
  • hypot(x, y)
  • __nv_fp128_hypot(x, y)
  • sin(x)
  • hsin(x)
  • hsin(x)
  • sinf(x)
  • sin(x)
  • __nv_fp128_sin(x)
  • cos(x)
  • hcos(x)
  • hcos(x)
  • cosf(x)
  • cos(x)
  • __nv_fp128_cos(x)
  • tan(x)
  • tanf(x)
  • tan(x)
  • __nv_fp128_tan(x)
  • asin(x)
  • asinf(x)
  • asin(x)
  • __nv_fp128_asin(x)
  • acos(x)
  • acosf(x)
  • acos(x)
  • __nv_fp128_acos(x)
  • atan(x)
  • atanf(x)
  • atan(x)
  • __nv_fp128_atan(x)
  • atan2(y, x)
  • atan2f(y, x)
  • atan2(y, x)
  • sinh(x)
  • sinhf(x)
  • sinh(x)
  • __nv_fp128_sinh(x)
  • cosh(x)
  • coshf(x)
  • cosh(x)
  • __nv_fp128_cosh(x)
  • tanh(x)
  • htanh(x)
  • htanh(x)
  • tanhf(x)
  • tanh(x)
  • __nv_fp128_tanh(x)
  • asinh(x)
  • asinhf(x)
  • asinh(x)
  • __nv_fp128_asinh(x)
  • acosh(x)
  • acoshf(x)
  • acosh(x)
  • __nv_fp128_acosh(x)
  • atanh(x)
  • atanhf(x)
  • atanh(x)
  • __nv_fp128_atanh(x)
  • erf(x)
  • erff(x)
  • erf(x)
  • erfc(x)
  • erfcf(x)
  • erfc(x)
  • tgamma(x)
  • tgammaf(x)
  • tgamma(x)
  • lgamma(x)
  • lgammaf(x)
  • lgamma(x)
  • ceil(x)
  • hceil(x)
  • hceil(x)
  • ceilf(x)
  • ceil(x)
  • __nv_fp128_ceil(x)
  • floor(x)
  • hfloor(x)
  • hfloor(x)
  • floorf(x)
  • floor(x)
  • __nv_fp128_floor(x)
  • trunc(x)
  • htrunc(x)
  • htrunc(x)
  • truncf(x)
  • trunc(x)
  • __nv_fp128_trunc(x)
  • round(x)
  • roundf(x)
  • round(x)
  • __nv_fp128_round(x)
  • nearbyint(x)
  • nearbyintf(x)
  • nearbyint(x)
  • rint(x)
  • hrint(x)
  • hrint(x)
  • rintf(x)
  • rint(x)
  • __nv_fp128_rint(x)
  • lrintf(x)
  • lrint(x)
  • llrintf(x)
  • llrint(x)
  • lroundf(x)
  • lround(x)
  • llroundf(x)
  • llround(x)
  • frexp(x, exp)
  • frexpf(x, exp)
  • frexp(x, exp)
  • __nv_fp128_frexp(x, nptr)
  • ldexp(x, n)
  • ldexpf(x, n)
  • ldexp(x, n)
  • __nv_fp128_ldexp(x, n)
  • modf(x, iptr)
  • modff(x, iptr)
  • modf(x, iptr)
  • __nv_fp128_modf(x, iptr)
  • scalbn(x, n)
  • scalbnf(x, n)
  • scalbn(x, n)
  • scalblnf(x, n)
  • scalbln(x, n)
  • ilogb(x)
  • ilogbf(x)
  • ilogb(x)
  • __nv_fp128_ilogb(x)
  • logb(x)
  • logbf(x)
  • logb(x)
  • nextafter(x, y)
  • nextafterf(x, y)
  • nextafter(x, y)
  • copysign(x, y)
  • copysignf(x, y)
  • copysign(x, y)
  • __nv_fp128_copysign(x, y)
  • fpclassify(x)
  • isfinite(x)
  • isfinite(x)
  • __hisinf(x)
  • __hisinf(x)
  • isinf(x)
  • isinf(x)
  • __hisnan(x)
  • __hisnan(x)
  • isnan(x)
  • isnan(x)
  • __nv_fp128_isnan(x)
  • signbit(x)
  • signbit(x)
  • signbit(x)
  • isgreater(x, y)
  • __hgt(x, y)
  • __hgt(x, y)
  • isgreaterequal(x, y)
  • __hge(x, y)
  • __hge(x, y)
  • isless(x, y)
  • __hlt(x, y)
  • __hlt(x, y)
  • islessequal(x, y)
  • __hle(x, y)
  • __hle(x, y)
  • islessgreater(x, y)
  • __hne(x, y)
  • __hne(x, y)
  • isunordered(x, y)
  • __nv_fp128_isunordered(x, y)
  • fdividef(x, y)
  • exp10f(x)
  • exp10(x)
  • norm3df(x, y, z)
  • norm3d(x, y, z)
  • norm4df(x, y, z, t)
  • norm4d(x, y, z, t)
  • normf(dim, p)
  • norm(dim, p)
  • rsqrtf(x)
  • rsqrt(x)
  • rcbrtf(x)
  • rcbrt(x)
  • rhypotf(x, y)
  • rhypot(x, y)
  • rnorm3df(x, y, z)
  • rnorm3d(x, y, z)
  • rnorm4df(x, y, z, t)
  • rnorm4d(x, y, z, t)
  • rnormf(dim, p)
  • rnorm(dim, p)
  • cospif(x)
  • cospi(x)
  • sinpif(x)
  • sinpi(x)
  • sincospif(x, sptr, cptr)
  • sincospi(x, sptr, cptr)
  • normcdff(x)
  • normcdf(x)
  • normcdfinvf(x)
  • normcdfinv(x)
  • erfcinvf(x)
  • erfcinv(x)
  • erfcxf(x)
  • erfcx(x)
  • erfinvf(x)
  • erfinv(x)
  • cyl_bessel_i0f(x)
  • cyl_bessel_i0(x)
  • cyl_bessel_i1f(x)
  • cyl_bessel_i1(x)
  • j0f(x)
  • j0(x)
  • j1f(x)
  • j1(x)
  • jnf(n, x)
  • jn(n, x)
  • y0f(x)
  • y0(x)
  • y1f(x)
  • y1(x)
  • ynf(n, x)
  • yn(n, x)
  • hrcp(x)
  • hrcp(x)
  • hexp10(x)
  • hexp10(x)
  • __nv_fp128_exp10(x)
  • hrsqrt(x)
  • hrsqrt(x)
  • htanh_approx(x)
  • htanh_approx(x)
  • --use_fast_math 的影响
  • _fadd[rn,rz,ru,rd](x, y)
  • _dadd[rn,rz,ru,rd](x, y)
  • _fsub[rn,rz,ru,rd](x, y)
  • _dsub[rn,rz,ru,rd](x, y)
  • _fmul[rn,rz,ru,rd](x, y)
  • _dmul[rn,rz,ru,rd](x, y)
  • _fma[rn,rz,ru,rd](x, y, z)
  • _fdiv[rn,rz,ru,rd](x, y)
  • _ddiv[rn,rz,ru,rd](x, y)
  • __frcp_[rn,rz,ru,rd](x)
  • __drcp_[rn,rz,ru,rd](x)
  • __fsqrt_[rn,rz,ru,rd](x)
  • __dsqrt_[rn,rz,ru,rd](x)
  • __fdividef(x, y)
  • __frsqrt_rn(x)
  • __expf(x)
  • __exp10f(x)
  • __powf(x, y)
  • __logf(x)
  • __log2f(x)
  • __log10f(x)
  • __sinf(x)
  • __cosf(x)
  • __sincosf(x, sptr, cptr)
  • __tanf(x)
  • __tanhf(x)
  • sincosf(x, sptr, cptr)
  • Precision & Performance: Floating Point and IEEE 754 Compliance for NVIDIA GPUs
  • What every computer scientist should know about floating-point arithmetic
  • The pitfalls of verifying floating-point computations
  • Do Developers Understand IEEE Floating Point?
Prev
5.4 C/C++ 语言扩展(C/C++ Language Extensions)
Next
5.6 设备可调用 API 与内建函数(Device-Callable APIs and Intrinsics)