tilelang.ascend.language.math¶

Ascend scalar math expressed with common TIR operations.

Functions¶

rsqrt(x, *[, dtype])

Compose scalar float32 reciprocal sqrt, preserving existing VF intrinsics.

tilelang.ascend.language.math.rsqrt(x, *, dtype=None)¶

Compose scalar float32 reciprocal sqrt, preserving existing VF intrinsics.

As with the common math wrappers, the input determines the result dtype. The optional dtype keyword is accepted for the same frontend contract.