__habs

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 absolute value of the input data.

Prototype

1
inline half __habs(half x)
1
inline bfloat16_t __habs(bfloat16_t x)

Parameters

Table 1 Parameters

Parameter

Input/Output

Description

x

Input

Source operand.

Returns

Absolute value of the input data.

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

Restrictions

For SIMT programming, this API is not supported.

Header File to Be Included

To use the half API, the simt_api/asc_fp16.h header file must be included. To use the bfloat16_t API, the simt_api/asc_bf16.h header file must be included.

1
#include "simt_api/asc_fp16.h"
1
#include "simt_api/asc_bf16.h"

Examples

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