tilelang.language.fastmath

Fast math operations exposed on the TileLang language surface.

Functions

fast_rcp(x)

Approximate reciprocal using CUDA fast reciprocal.

Module Contents

tilelang.language.fastmath.fast_rcp(x)

Approximate reciprocal using CUDA fast reciprocal.

Parameters:

x (tvm.tirx.PrimExpr)

Return type:

tvm.tirx.PrimExpr