hlog2
产 品 支 持 情 况
- Ascend 950PR/Ascend 950DT:支 持
- Atlas A3 训 练 系 列 产 品/Atlas A3 推 理 系 列 产 品:不 支 持
- Atlas A2 训 练 系 列 产 品/Atlas A2 推 理 系 列 产 品:不 支 持
- Atlas 200I/500 A2 推 理 产 品:不 支 持
- Atlas 推 理 系 列 产 品AI Core:不 支 持
- Atlas 推 理 系 列 产 品Vector Core:不 支 持
- Atlas 训 练 系 列 产 品:不 支 持
功 能 说 明
获 取 以2为 底,输 入 数 据 的 对 数。
函 数 原 型
Text
inline bfloat16_t hlog2(bfloat16_t x)
参 数 说 明
表1 参 数 说 明
| 参 数 名 | 输 入/输 出 | 描 述 |
|---|---|---|
| x | 输 入 | 源 操 作 数。 |
返 回 值 说 明
以2为 底 的x的 对 数。本 接 口 受 全 局 饱 和 模 式 影 响,特 殊 值 如 下:
| x值 | 非 饱 和 模 式 返 回 值 | 饱 和 模 式 返 回 值 |
|---|---|---|
| inf | inf | ASCRT_MAX_NORMAL_BF16(饱 和 到bfloat16_t最 大 有 限 值) |
| -inf, nan | nan | 0(nan被 饱 和 为0) |
| 0 | -inf | -ASCRT_MAX_NORMAL_BF16 |
| x小 于0 | nan | 0 |
约 束 说 明
无
需 要 包 含 的 头 文 件
使 用bfloat16_t类 型 接 口 需 要 包 含"simt_api/asc_bf16.h"头 文 件。
Text
#include "simt_api/asc_bf16.h"
调 用 示 例
SIMT编 程 场 景:
Text__global__ __launch_bounds__(1024) void KernelLog2(bfloat16_t* dst, bfloat16_t* x) { int idx = threadIdx.x + blockIdx.x * blockDim.x; dst[idx] = hlog2(x[idx]); }SIMD与SIMT混 合 编 程 场 景:
Text__simt_vf__ __launch_bounds__(1024) inline void KernelLog2(__gm__ bfloat16_t* dst, __gm__ bfloat16_t* x) { int idx = threadIdx.x + blockIdx.x * blockDim.x; dst[idx] = hlog2(x[idx]); }