4.6. Half2 算术函数
要使用这些函数，请在程序中包含头文件 cuda_fp16.h。
函数
- __host__ __device__ __half2 __h2div(const __half2 a, const __half2 b)
-
以舍入到最近的偶数模式执行
half2向量除法。 - __host__ __device__ __half2 __habs2(const __half2 a)
-
计算输入
half2数字的两个半部分的绝对值，并返回结果。 - __host__ __device__ __half2 __hadd2(const __half2 a, const __half2 b)
-
以舍入到最近的偶数模式执行
half2向量加法。 - __host__ __device__ __half2 __hadd2_rn(const __half2 a, const __half2 b)
-
以舍入到最近的偶数模式执行
half2向量加法。 - __host__ __device__ __half2 __hadd2_sat(const __half2 a, const __half2 b)
-
以舍入到最近的偶数模式执行
half2向量加法，结果饱和到 [0.0, 1.0]。 - __device__ __half2 __hcmadd(const __half2 a, const __half2 b, const __half2 c)
-
执行快速复数乘法累加。
- __device__ __half2 __hfma2(const __half2 a, const __half2 b, const __half2 c)
-
以舍入到最近的偶数模式执行
half2向量融合乘法累加。 - __device__ __half2 __hfma2_relu(const __half2 a, const __half2 b, const __half2 c)
-
以舍入到最近的偶数模式执行
half2向量融合乘法累加，并使用 ReLU 饱和。 - __device__ __half2 __hfma2_sat(const __half2 a, const __half2 b, const __half2 c)
-
以舍入到最近的偶数模式执行
half2向量融合乘法累加，结果饱和到 [0.0, 1.0]。 - __host__ __device__ __half2 __hmul2(const __half2 a, const __half2 b)
-
以舍入到最近的偶数模式执行
half2向量乘法。 - __host__ __device__ __half2 __hmul2_rn(const __half2 a, const __half2 b)
-
以舍入到最近的偶数模式执行
half2向量乘法。 - __host__ __device__ __half2 __hmul2_sat(const __half2 a, const __half2 b)
-
以舍入到最近的偶数模式执行
half2向量乘法，结果饱和到 [0.0, 1.0]。 - __host__ __device__ __half2 __hneg2(const __half2 a)
-
对输入
half2数字的两个半部分取反，并返回结果。 - __host__ __device__ __half2 __hsub2(const __half2 a, const __half2 b)
-
以舍入到最近的偶数模式执行
half2向量减法。 - __host__ __device__ __half2 __hsub2_rn(const __half2 a, const __half2 b)
-
以舍入到最近的偶数模式执行
half2向量减法。 - __host__ __device__ __half2 __hsub2_sat(const __half2 a, const __half2 b)
-
以舍入到最近的偶数模式执行
half2向量减法，结果饱和到 [0.0, 1.0]。 - __device__ __half2 atomicAdd(__half2 *const address, const __half2 val)
-
向量加法：将
val加到存储在全局或共享内存address的值，并将此值写回address。 - __host__ __device__ __half2 operator*(const __half2 &lh, const __half2 &rh)
-
执行 packed
half乘法运算。 - __host__ __device__ __half2 & operator*=(__half2 &lh, const __half2 &rh)
-
使用乘法运算执行 packed
half复合赋值。 - __host__ __device__ __half2 operator+(const __half2 &h)
-
实现 packed
half一元加运算符，返回输入值。 - __host__ __device__ __half2 operator+(const __half2 &lh, const __half2 &rh)
-
执行 packed
half加法运算。 - __host__ __device__ __half2 operator++(__half2 &h, const int ignored)
-
执行 packed
half后缀递增运算。 - __host__ __device__ __half2 & operator++(__half2 &h)
-
执行 packed
half前缀递增运算。 - __host__ __device__ __half2 & operator+=(__half2 &lh, const __half2 &rh)
-
使用加法运算执行 packed
half复合赋值。 - __host__ __device__ __half2 operator-(const __half2 &h)
-
实现 packed
half一元减运算符。 - __host__ __device__ __half2 operator-(const __half2 &lh, const __half2 &rh)
-
执行 packed
half减法运算。 - __host__ __device__ __half2 & operator–(__half2 &h)
-
执行 packed
half前缀递减运算。 - __host__ __device__ __half2 operator–(__half2 &h, const int ignored)
-
执行 packed
half后缀递减运算。 - __host__ __device__ __half2 & operator-=(__half2 &lh, const __half2 &rh)
-
使用减法运算执行 packed
half复合赋值。 - __host__ __device__ __half2 operator/(const __half2 &lh, const __half2 &rh)
-
执行 packed
half除法运算。 - __host__ __device__ __half2 & operator/=(__half2 &lh, const __half2 &rh)
-
使用除法运算执行 packed
half复合赋值。
4.6.1. 函数
-
__host__ __device__ __half2 __h2div(const __half2 a, const __half2 b)
-
以舍入到最近的偶数模式执行
half2向量除法。以舍入到最近的偶数模式，将
half2输入向量a除以输入向量b。- 参数
-
a – [in] - half2。只读。
b – [in] - half2。只读。
- 返回值
-
half2
a除以b的元素级除法结果。
-
__host__ __device__ __half2 __habs2(const __half2 a)
-
计算输入
half2数字的两个半部分的绝对值，并返回结果。计算输入
half2数字的两个半部分的绝对值，并返回结果。另请参阅
__habs(__half) 以了解更多详细信息。
- 参数
-
a – [in] - half2。只读。
- 返回值
-
half2
返回
a，其中包含两个半部分的绝对值。
-
__host__ __device__ __half2 __hadd2(const __half2 a, const __half2 b)
-
以舍入到最近的偶数模式执行
half2向量加法。以舍入到最近的偶数模式，执行输入
a和b的half2向量加法。- 参数
-
a – [in] - half2。只读。
b – [in] - half2。只读。
- 返回值
-
half2
向量
a和b的和。
-
__host__ __device__ __half2 __hadd2_rn(const __half2 a, const __half2 b)
-
以舍入到最近的偶数模式执行
half2向量加法。以舍入到最近的偶数模式，执行输入
a和b的half2向量加法。防止将 mul+add 浮点收缩为 fma。- 参数
-
a – [in] - half2。只读。
b – [in] - half2。只读。
- 返回值
-
half2
向量
a和b的和。
-
__host__ __device__ __half2 __hadd2_sat(const __half2 a, const __half2 b)
-
以舍入到最近的偶数模式执行
half2向量加法，结果饱和到 [0.0, 1.0]。以舍入到最近的偶数模式，执行输入
a和b的half2向量加法，并将结果钳位到 [0.0, 1.0] 范围。NaN 结果被刷新为 +0.0。- 参数
-
a – [in] - half2。只读。
b – [in] - half2。只读。
- 返回值
-
half2
a和b的和，考虑了饱和。
-
__device__ __half2 __hcmadd(const __half2 a, const __half2 b, const __half2 c)
-
执行快速复数乘法累加。
将向量
half2输入对a、b和c解释为half精度复数：(a.x + I*a.y)、(b.x + I*b.y)、(c.x + I*c.y)，并以简单方式执行复数乘法累加运算：a*b + c，即：((a.x*b.x + c.x) - a.y*b.y) + I*((a.x*b.y + c.y) + a.y*b.x)- 参数
-
a – [in] - half2。只读。
b – [in] - half2。只读。
c – [in] - half2。只读。
- 返回值
-
half2
复数
a、b和c的复数乘法累加运算结果__half2 result = __hcmadd(a, b, c) 在数值上与以下结果一致
result.x = __hfma(-a.y, b.y, __hfma(a.x, b.x, c.x))
result.y = __hfma( a.y, b.x, __hfma(a.x, b.y, c.y))
-
__device__ __half2 __hfma2(const __half2 a, const __half2 b, const __half2 c)
-
以舍入到最近的偶数模式执行
half2向量融合乘法累加。以舍入到最近的偶数模式，对输入
a和b执行half2向量乘法，然后将结果与half2向量c相加，并将结果舍入一次。- 参数
-
a – [in] - half2。只读。
b – [in] - half2。只读。
c – [in] - half2。只读。
- 返回值
-
half2
向量
a、b和c的元素级融合乘法累加运算结果。
-
__device__ __half2 __hfma2_relu(const __half2 a, const __half2 b, const __half2 c)
-
以舍入到最近的偶数模式执行
half2向量融合乘法累加，并使用 ReLU 饱和。以舍入到最近的偶数模式，对输入
a和b执行half2向量乘法，然后将结果与half2向量c相加，并将结果舍入一次。然后将负结果钳位为 0。NaN 结果转换为规范 NaN。- 参数
-
a – [in] - half2。只读。
b – [in] - half2。只读。
c – [in] - half2。只读。
- 返回值
-
half2
向量
a、b和c的元素级融合乘法累加运算结果，带有 ReLU 饱和。
-
__device__ __half2 __hfma2_sat(const __half2 a, const __half2 b, const __half2 c)
-
以舍入到最近的偶数模式执行
half2向量融合乘法累加，结果饱和到 [0.0, 1.0]。以舍入到最近的偶数模式，对输入
a和b执行half2向量乘法，然后将结果与half2向量c相加，并将结果舍入一次，并将结果钳位到 [0.0, 1.0] 范围。NaN 结果被刷新为 +0.0。- 参数
-
a – [in] - half2。只读。
b – [in] - half2。只读。
c – [in] - half2。只读。
- 返回值
-
half2
向量
a、b和c的元素级融合乘法累加运算结果，考虑了饱和。
-
__host__ __device__ __half2 __hmul2(const __half2 a, const __half2 b)
-
以舍入到最近的偶数模式执行
half2向量乘法。以舍入到最近的偶数模式，执行输入
a和b的half2向量乘法。- 参数
-
a – [in] - half2。只读。
b – [in] - half2。只读。
- 返回值
-
half2
向量
a和b的元素级乘法结果。
-
__host__ __device__ __half2 __hmul2_rn(const __half2 a, const __half2 b)
-
以舍入到最近的偶数模式执行
half2向量乘法。以舍入到最近的偶数模式，执行输入
a和b的half2向量乘法。防止将 mul+add 或 sub 浮点收缩为 fma。- 参数
-
a – [in] - half2。只读。
b – [in] - half2。只读。
- 返回值
-
half2
向量
a和b的元素级乘法结果。
-
__host__ __device__ __half2 __hmul2_sat(const __half2 a, const __half2 b)
-
以舍入到最近的偶数模式执行
half2向量乘法，结果饱和到 [0.0, 1.0]。以舍入到最近的偶数模式，执行输入
a和b的half2向量乘法，并将结果钳位到 [0.0, 1.0] 范围。NaN 结果被刷新为 +0.0。- 参数
-
a – [in] - half2。只读。
b – [in] - half2。只读。
- 返回值
-
half2
向量
a和b的元素级乘法结果，考虑了饱和。
-
__host__ __device__ __half2 __hneg2(const __half2 a)
-
对输入
half2数字的两个半部分取反，并返回结果。对输入
half2数字a的两个半部分取反，并返回结果。另请参阅
有关更多详细信息，请参阅 __hneg(__half)。
- 参数
-
a – [in] - half2。只读。
- 返回值
-
half2
返回取反两个半精度的
a。
-
__host__ __device__ __half2 __hsub2(const __half2 a, const __half2 b)
-
以舍入到最近的偶数模式执行
half2向量减法。以舍入到最近偶数模式从输入向量
a中减去half2输入向量b。- 参数
-
a – [in] - half2。只读。
b – [in] - half2。只读。
- 返回值
-
half2
向量
b从a的减法。
-
__host__ __device__ __half2 __hsub2_rn(const __half2 a, const __half2 b)
-
以舍入到最近的偶数模式执行
half2向量减法。以舍入到最近偶数模式从输入向量
a中减去half2输入向量b。阻止将 mul+sub 的浮点收缩优化为 fma。- 参数
-
a – [in] - half2。只读。
b – [in] - half2。只读。
- 返回值
-
half2
向量
b从a的减法。
-
__host__ __device__ __half2 __hsub2_sat(const __half2 a, const __half2 b)
-
以舍入到最近的偶数模式执行
half2向量减法，结果饱和到 [0.0, 1.0]。以舍入到最近偶数模式从输入向量
a中减去half2输入向量b，并将结果钳位到范围 [0.0, 1.0]。NaN 结果将被刷新为 +0.0。- 参数
-
a – [in] - half2。只读。
b – [in] - half2。只读。
- 返回值
-
half2
向量
b从a的减法，考虑了饱和。
-
__device__ __half2 atomicAdd(__half2 *const address, const __half2 val)
-
向量加法
val到存储在全局或共享内存的address的值，并将此值写回address。对于两个
__half元素中的每一个，加法操作的原子性是单独保证的；不能保证整个__half2作为单个 32 位访问是原子的。address的位置必须在全局或共享内存中。否则，此操作具有未定义的行为。此操作由计算能力为 6.x 和更高的设备原生支持，较旧的设备使用模拟路径。注意
有关此功能的更多详细信息，请参阅 CUDA C++ 编程指南中的原子函数部分。
- 参数
-
address – [in] - half2*。全局或共享内存中的地址。
val – [in] - half2。要添加的值。
- 返回值
-
half2
从
address读取的旧值。