rnorm4df
产品支持情况
产品 |
是否支持 |
|---|---|
Atlas 350 加速卡 |
√ |
x |
|
x |
|
x |
|
x |
|
x |
|
x |
功能说明
获取输入数据a、b、c、d的平方和a^2 + b^2 + c^2 + d^2的平方根的倒数。

函数原型
1 | inline float rnorm4df(float a, float b, float c, float d) |
参数说明
参数名 |
输入/输出 |
描述 |
|---|---|---|
a |
输入 |
源操作数。 |
b |
输入 |
源操作数。 |
c |
输入 |
源操作数。 |
d |
输入 |
源操作数。 |
返回值说明
a^2 + b^2 + c^2 + d^2的平方根的倒数。
- 若a^2 + b^2 + c^2 + d^2的平方根超出float最大范围,返回值为0。
- 若a^2 + b^2 + c^2 + d^2平方根的倒数超出float最大范围,返回值为inf。
- 若a、b、c、d都为0,返回值为inf。
- 若a、b、c、d任意一个或多个为±inf,返回值为0。
- 若a、b、c、d任意一个或多个为nan同时不是±inf,返回值为nan。
约束说明
SIMT编程场景当前不支持使用该接口。
需要包含的头文件
使用该接口需要包含"simt_api/math_functions.h"头文件。
1 | #include "simt_api/math_functions.h" |
调用示例
SIMD与SIMT混合编程场景:
1 2 3 4 5 | __simt_vf__ __launch_bounds__(1024) inline void KernelRnorm4d(__gm__ float* dst, __gm__ float* a, __gm__ float* b, __gm__ float* c, __gm__ float* d) { int idx = threadIdx.x + blockIdx.x * blockDim.x; dst[idx] = rnorm4df(a[idx], b[idx], c[idx], d[idx]); } |
父主题: 数学函数