triton.language.sqrt

triton.language.sqrt(x, builder=None)

Computes the element-wise square root of x

Parameters

• x (Block) – the input values

• builder (triton.ir.builder, optional from within JIT'ed functions) – IR builder to generate code into