rsqrtf

Applicability

Product

Supported

Atlas 350 Accelerator Card

Atlas A3 training product/Atlas A3 inference product

x

Atlas A2 training product/Atlas A2 inference product

x

Atlas 200I/500 A2 inference product

x

Atlas inference product AI Core

x

Atlas inference product Vector Core

x

Atlas training product

x

Function Usage

Obtains the reciprocal of the square root of the input data x.

Prototype

1
inline float rsqrtf(float x)

Parameters

Table 1 Parameters

Parameter

Input/Output

Description

x

Input

Source operand, input data.

Returns

Reciprocal of the square root of the input data.

  • If x is 0, the return value is inf.
  • If x is inf, the return value is 0.
  • If x is -inf, the return value is nan.
  • If x is nan, the return value is nan.

Restrictions

The input data x supported by this API must be greater than or equal to 0. Otherwise, the return value is nan.

For SIMT programming, this API is not supported.

Header File to Be Included

To use this API, the simt_api/math_functions.h header file must be included.

1
#include "simt_api/math_functions.h"

Examples

For SIMD and SIMT programming:
1
2
3
4
5
__simt_vf__ __launch_bounds__(1024) inline void KernelRsqrt(__gm__ float* dst, __gm__ float* x)
{
    int idx = threadIdx.x + blockIdx.x * blockDim.x;
    dst[idx] = rsqrtf(x[idx]);
}