__saturatef
产品支持情况
产品 |
是否支持 |
|---|---|
Atlas 350 加速卡 |
√ |
x |
|
x |
|
x |
|
x |
|
x |
|
x |
功能说明
将输入数据转换至[0.0, 1.0]区间。具体来说,若输入数据大于或等于1,则结果为1,若输入数据小于或等于0,则结果为0,若输入数据在[0.0, 1.0]区间,则结果为原输入数据。
函数原型
1 | float __saturatef(float x) |
参数说明
参数名 |
输入/输出 |
描述 |
|---|---|---|
x |
输入 |
源操作数。 |
返回值说明
输入数据转换至[0.0, 1.0]区间的结果。
- 当x≤0时,返回值为0。
- 当x≥1时,返回值为1。
- 当0<x<1时,返回值为x。
- 当x为nan时,返回值为0。
约束说明
无
需要包含的头文件
使用该接口需要包含"simt_api/device_functions.h"头文件。
1 | #include "simt_api/device_functions.h" |
调用示例
- SIMT编程场景:
1 2 3 4 5
__global__ __launch_bounds__(1024) void KernelSaturatef(float* dst, float* x) { int idx = threadIdx.x + blockIdx.x * blockDim.x; dst[idx] = __saturatef(x[idx]); }
- SIMD与SIMT混合编程场景:
1 2 3 4 5
__simt_vf__ __launch_bounds__(1024) inline void KernelSaturatef(__gm__ float* dst, __gm__ float* x) { int idx = threadIdx.x + blockIdx.x * blockDim.x; dst[idx] = __saturatef(x[idx]); }
父主题: float类型数学库函数