5.3. Bfloat16 比较函数

要使用这些函数,请在程序中包含头文件cuda_bf16.h

Functions

__host__ __device__ bool __heq(const __nv_bfloat16 a, const __nv_bfloat16 b)

执行nv_bfloat16的if-equal比较。

__host__ __device__ bool __hequ(const __nv_bfloat16 a, const __nv_bfloat16 b)

执行nv_bfloat16无序等于比较。

__host__ __device__ bool __hge(const __nv_bfloat16 a, const __nv_bfloat16 b)

执行nv_bfloat16大于等于比较。

__host__ __device__ bool __hgeu(const __nv_bfloat16 a, const __nv_bfloat16 b)

执行nv_bfloat16无序大于等于比较。

__host__ __device__ bool __hgt(const __nv_bfloat16 a, const __nv_bfloat16 b)

执行nv_bfloat16大于比较。

__host__ __device__ bool __hgtu(const __nv_bfloat16 a, const __nv_bfloat16 b)

执行nv_bfloat16无序大于比较。

__host__ __device__ int __hisinf(const __nv_bfloat16 a)

检查输入的nv_bfloat16数值是否为无穷大。

__host__ __device__ bool __hisnan(const __nv_bfloat16 a)

判断nv_bfloat16参数是否为NaN。

__host__ __device__ bool __hle(const __nv_bfloat16 a, const __nv_bfloat16 b)

执行nv_bfloat16小于等于比较。

__host__ __device__ bool __hleu(const __nv_bfloat16 a, const __nv_bfloat16 b)

执行nv_bfloat16无序小于等于比较。

__host__ __device__ bool __hlt(const __nv_bfloat16 a, const __nv_bfloat16 b)

执行nv_bfloat16小于比较。

__host__ __device__ bool __hltu(const __nv_bfloat16 a, const __nv_bfloat16 b)

执行nv_bfloat16无序小于比较。

__host__ __device__ __nv_bfloat16 __hmax(const __nv_bfloat16 a, const __nv_bfloat16 b)

计算两个输入值的nv_bfloat16最大值。

__host__ __device__ __nv_bfloat16 __hmax_nan(const __nv_bfloat16 a, const __nv_bfloat16 b)

计算两个输入值的nv_bfloat16最大值,NaN值直接传递。

__host__ __device__ __nv_bfloat16 __hmin(const __nv_bfloat16 a, const __nv_bfloat16 b)

计算两个输入值的nv_bfloat16最小值。

__host__ __device__ __nv_bfloat16 __hmin_nan(const __nv_bfloat16 a, const __nv_bfloat16 b)

计算两个输入值的nv_bfloat16最小值,NaN值直接透传。

__host__ __device__ bool __hne(const __nv_bfloat16 a, const __nv_bfloat16 b)

执行nv_bfloat16不相等比较。

__host__ __device__ bool __hneu(const __nv_bfloat16 a, const __nv_bfloat16 b)

执行nv_bfloat16无序不等于比较。

__host__ __device__ bool operator!=(const __nv_bfloat16 &lh, const __nv_bfloat16 &rh)

执行nv_bfloat16无序不等于比较操作。

__host__ __device__ bool operator<(const __nv_bfloat16 &lh, const __nv_bfloat16 &rh)

执行nv_bfloat16有序小于比较操作。

__host__ __device__ bool operator<=(const __nv_bfloat16 &lh, const __nv_bfloat16 &rh)

执行nv_bfloat16有序小于等于比较操作。

__host__ __device__ bool operator==(const __nv_bfloat16 &lh, const __nv_bfloat16 &rh)

执行nv_bfloat16有序比较相等操作。

__host__ __device__ bool operator>(const __nv_bfloat16 &lh, const __nv_bfloat16 &rh)

执行nv_bfloat16有序大于比较操作。

__host__ __device__ bool operator>=(const __nv_bfloat16 &lh, const __nv_bfloat16 &rh)

执行nv_bfloat16有序大于等于比较操作。

5.3.1. 函数

__host__ __device__ bool __heq(const __nv_bfloat16 a, const __nv_bfloat16 b)

执行nv_bfloat16的if-equal比较。

对输入ab执行nv_bfloat16类型的相等比较。NaN输入会产生false结果。

Parameters
  • a[输入] - nv_bfloat16。仅用于读取。

  • b[输入] - nv_bfloat16。仅用于读取。

Returns

布尔值

  • ab 进行相等比较的布尔结果。

__host__ __device__ bool __hequ(const __nv_bfloat16 a, const __nv_bfloat16 b)

执行nv_bfloat16无序等于比较。

对输入ab执行nv_bfloat16类型的相等比较。NaN输入会产生true结果。

Parameters
  • a[输入] - nv_bfloat16。仅用于读取。

  • b[输入] - nv_bfloat16。仅用于读取。

Returns

布尔值

  • ab 的无序相等比较的布尔结果。

__host__ __device__ bool __hge(const __nv_bfloat16 a, const __nv_bfloat16 b)

执行nv_bfloat16大于等于比较。

对输入ab执行nv_bfloat16大于等于比较。NaN输入会产生false结果。

Parameters
  • a[输入] - nv_bfloat16。仅用于读取。

  • b[输入] - nv_bfloat16。仅用于读取。

Returns

布尔值

  • ab 的大于等于比较的布尔结果。

__host__ __device__ bool __hgeu(const __nv_bfloat16 a, const __nv_bfloat16 b)

执行nv_bfloat16无序大于等于比较。

对输入ab执行nv_bfloat16大于等于比较。NaN输入会产生true结果。

Parameters
  • a[输入] - nv_bfloat16。仅用于读取。

  • b[输入] - nv_bfloat16。仅用于读取。

Returns

布尔值

  • ab 的无序大于等于比较的布尔结果。

__host__ __device__ bool __hgt(const __nv_bfloat16 a, const __nv_bfloat16 b)

执行nv_bfloat16大于比较。

对输入ab执行nv_bfloat16大于比较。NaN输入会产生false结果。

Parameters
  • a[in] - nv_bfloat16。仅用于读取。

  • b[输入] - nv_bfloat16。仅用于读取。

Returns

布尔值

  • ab 进行大于比较的布尔结果。

__host__ __device__ bool __hgtu(const __nv_bfloat16 a, const __nv_bfloat16 b)

执行nv_bfloat16无序大于比较。

对输入ab执行nv_bfloat16大于比较。NaN输入会产生true结果。

Parameters
  • a[输入] - nv_bfloat16。仅用于读取。

  • b[输入] - nv_bfloat16。仅用于读取。

Returns

布尔值

  • ab 的无序大于比较的布尔结果。

__host__ __device__ int __hisinf(const __nv_bfloat16 a)

检查输入的nv_bfloat16数值是否为无穷大。

检查输入的nv_bfloat16数字a是否为无穷大。

Parameters

a[输入] - nv_bfloat16。仅用于读取。

Returns

int

  • 如果 a 等于负无穷大,则返回 -1

  • 如果 a 等于正无穷大,则返回 1

  • 否则为0。

__host__ __device__ bool __hisnan(const __nv_bfloat16 a)

判断nv_bfloat16参数是否为NaN值。

判断nv_bfloat16a是否为NaN。

Parameters

a[输入] - nv_bfloat16。仅用于读取。

Returns

布尔值

  • 如果参数是NaN则返回true。

__host__ __device__ bool __hle(const __nv_bfloat16 a, const __nv_bfloat16 b)

执行nv_bfloat16小于等于比较。

对输入ab执行nv_bfloat16小于等于比较。NaN输入会产生false结果。

Parameters
  • a[输入] - nv_bfloat16。仅用于读取。

  • b[输入] - nv_bfloat16。仅用于读取。

Returns

布尔值

  • ab 的小于等于比较的布尔结果。

__host__ __device__ bool __hleu(const __nv_bfloat16 a, const __nv_bfloat16 b)

执行nv_bfloat16无序小于等于比较。

对输入ab执行nv_bfloat16小于等于比较。NaN输入会产生true结果。

Parameters
  • a[输入] - nv_bfloat16。仅用于读取。

  • b[输入] - nv_bfloat16。仅用于读取。

Returns

布尔值

  • ab进行无序小于等于比较的布尔结果。

__host__ __device__ bool __hlt(const __nv_bfloat16 a, const __nv_bfloat16 b)

执行nv_bfloat16小于比较。

对输入ab执行nv_bfloat16小于比较。NaN输入会产生false结果。

Parameters
  • a[输入] - nv_bfloat16。仅用于读取。

  • b[输入] - nv_bfloat16。仅用于读取。

Returns

布尔值

  • ab 小于比较的布尔结果。

__host__ __device__ bool __hltu(const __nv_bfloat16 a, const __nv_bfloat16 b)

执行nv_bfloat16无序小于比较。

对输入ab执行nv_bfloat16小于比较。NaN输入会产生true结果。

Parameters
  • a[输入] - nv_bfloat16。仅用于读取。

  • b[输入] - nv_bfloat16。仅用于读取。

Returns

布尔值

  • ab 的无序小于比较的布尔结果。

__host__ __device__ __nv_bfloat16 __hmax(const __nv_bfloat16 a, const __nv_bfloat16 b)

计算两个输入值的nv_bfloat16最大值。

计算 nv_bfloat16 的最大值 max(a, b),定义为 (a > b) ? a : b

  • 如果任一输入为NaN,则返回另一个输入。

  • 如果两个输入都是NaN,则返回规范NaN。

  • 如果两个输入值均为0.0,则+0.0 > -0.0

Parameters
  • a[输入] - nv_bfloat16。仅用于读取。

  • b[输入] - nv_bfloat16。仅用于读取。

Returns

nv_bfloat16

__host__ __device__ __nv_bfloat16 __hmax_nan(const __nv_bfloat16 a, const __nv_bfloat16 b)

计算两个输入值的nv_bfloat16最大值,NaN值直接透传。

计算nv_bfloat16类型的最大值max(a, b),定义为(a > b) ? a : b

  • 如果任一输入为NaN,则返回标准NaN。

  • 如果两个输入值均为0.0,则+0.0 > -0.0

Parameters
  • a[输入] - nv_bfloat16。仅用于读取。

  • b[输入] - nv_bfloat16。仅用于读取。

Returns

nv_bfloat16

__host__ __device__ __nv_bfloat16 __hmin(const __nv_bfloat16 a, const __nv_bfloat16 b)

计算两个输入值的nv_bfloat16最小值。

计算nv_bfloat16最小值(a, b),定义为(a < b) ? a : b

  • 如果任一输入为NaN,则返回另一个输入。

  • 如果两个输入都是NaN,则返回规范NaN。

  • 如果两个输入值均为0.0,则+0.0 > -0.0

Parameters
  • a[输入] - nv_bfloat16。仅用于读取。

  • b[输入] - nv_bfloat16。仅用于读取。

Returns

nv_bfloat16

__host__ __device__ __nv_bfloat16 __hmin_nan(const __nv_bfloat16 a, const __nv_bfloat16 b)

计算两个输入值的nv_bfloat16最小值,NaN值直接透传。

计算nv_bfloat16最小值(a, b),定义为(a < b) ? a : b

  • 如果任一输入为NaN,则返回标准NaN。

  • 如果两个输入值均为0.0,则+0.0 > -0.0

Parameters
  • a[输入] - nv_bfloat16。仅用于读取。

  • b[输入] - nv_bfloat16。仅用于读取。

Returns

nv_bfloat16

__host__ __device__ bool __hne(const __nv_bfloat16 a, const __nv_bfloat16 b)

执行nv_bfloat16不相等比较。

对输入ab执行nv_bfloat16不相等比较。NaN输入会产生false结果。

Parameters
  • a[输入] - nv_bfloat16。仅用于读取。

  • b[输入] - nv_bfloat16。仅用于读取。

Returns

布尔值

  • ab 的不等于比较的布尔结果。

__host__ __device__ bool __hneu(const __nv_bfloat16 a, const __nv_bfloat16 b)

执行nv_bfloat16无序不等于比较。

对输入ab执行nv_bfloat16不相等比较。NaN输入会产生true结果。

Parameters
  • a[输入] - nv_bfloat16。仅用于读取。

  • b[输入] - nv_bfloat16。仅用于读取。

Returns

布尔值

  • ab 的无序不相等比较的布尔结果。

__host__ __device__ bool operator!=(const __nv_bfloat16 &lh, const __nv_bfloat16 &rh)

执行nv_bfloat16无序不等于比较操作。

另请参阅 __hneu(__nv_bfloat16, __nv_bfloat16)

__host__ __device__ bool operator<(const __nv_bfloat16 &lh, const __nv_bfloat16 &rh)

执行nv_bfloat16有序小于比较操作。

另请参阅 __hlt(__nv_bfloat16, __nv_bfloat16)

__host__ __device__ bool operator<=(const __nv_bfloat16 &lh, const __nv_bfloat16 &rh)

执行nv_bfloat16有序小于等于比较操作。

另请参阅 __hle(__nv_bfloat16, __nv_bfloat16)

__host__ __device__ bool operator==(const __nv_bfloat16 &lh, const __nv_bfloat16 &rh)

执行nv_bfloat16有序比较相等操作。

另请参阅 __heq(__nv_bfloat16, __nv_bfloat16)

__host__ __device__ bool operator>(const __nv_bfloat16 &lh, const __nv_bfloat16 &rh)

执行nv_bfloat16有序大于比较操作。

另请参阅 __hgt(__nv_bfloat16, __nv_bfloat16)

__host__ __device__ bool operator>=(const __nv_bfloat16 &lh, const __nv_bfloat16 &rh)

执行nv_bfloat16有序大于等于比较操作。

另请参阅 __hge(__nv_bfloat16, __nv_bfloat16)