h2rcp
产品支持情况
产品 |
是否支持 |
|---|---|
Ascend 950PR/Ascend 950DT |
√ |
x |
|
x |
|
x |
|
x |
|
x |
|
x |
功能说明
获取输入数据x各元素的倒数。

函数原型
1 | inline half2 h2rcp(half2 x) |
参数说明
参数名 |
输入/输出 |
描述 |
|---|---|---|
x |
输入 |
源操作数,输入数据。 |
返回值说明
输入数据各元素的倒数。
- 当输入元素为0时,返回值为inf。
- 当输入元素为inf时,返回值为0。
- 当输入元素为nan时,返回值为nan。
约束说明
无
需要包含的头文件
使用half2类型接口需要包含"simt_api/asc_fp16.h"头文件。
1 | #include "simt_api/asc_fp16.h" |
调用示例
- SIMT编程场景:
1 2 3 4 5
__global__ __launch_bounds__(1024) void KernelRcp(half2* dst, half2* x) { int idx = threadIdx.x + blockIdx.x * blockDim.x; dst[idx] = h2rcp(x[idx]); }
- SIMD与SIMT混合编程场景:
1 2 3 4 5
__simt_vf__ __launch_bounds__(1024) inline void KernelRcp(__gm__ half2* dst, __gm__ half2* x) { int idx = threadIdx.x + blockIdx.x * blockDim.x; dst[idx] = h2rcp(x[idx]); }
父主题: half2类型数学库函数