triton.language.sqrt_rn

triton.language.sqrt_rn(x, _semantic=None)

Computes the element-wise precise square root (rounding to nearest wrt the IEEE standard) of x.

Parameters:

x (Block) – the input values

This function can also be called as a member function on tensor, as x.sqrt_rn() instead of sqrt_rn(x).