triton.language.extra.cann.libdevice

Libdevice Ops

acos	Computes the element-wise acos of x.
acosh	Computes the element-wise acosh of x.
asin	Computes the element-wise arcsine of x.
asinh	Computes the element-wise asinh of x.
atan	Computes the element-wise arctangent (inverse tangent) of the input tensor.
atan2	Computes the arctangent of arg0/arg1, using signs to determine the quadrant.
atanh	Computes the element-wise atanh of x.
copysign	Computes the element-wise copysign of x and y.
cosh	Computes the element-wise cosh of x.
cyl_bessel_i0	Computes the element-wise besseli0 (Modified Bessel function of the first kind, order 0).
div_rz	Computes the division with round-toward-zero mode.
erfinv	Computes the inverse error function.
expm1	Computes the element-wise expm1 of x.
fast_dividef	Computes a fast floating-point division.
fast_expf	Computes a fast exponential (e^x) of the input tensor.
float_as_int	Reinterprets the bits of a float32 value as an int32.
fmod	Computes the floating-point remainder of arg0 / arg1.
gamma	Computes the Gamma function using the Lanczos approximation.
hypot	Computes the element-wise hypot(Euclidean Distance) of x and y.
ilogb	Returns the integer binary exponent of the input tensor.
isinf	Tests whether each element of the input tensor is infinity.
isnan	Tests whether each element of the input tensor is NaN.
ldexp	Computes x * 2^exp from a mantissa and an exponent.
lgamma	Computes the natural logarithm of the absolute value of the Gamma function.
log10	Computes the element-wise base-10 logarithm of x.
log1p	Computes the element-wise natural logarithm of (1 + x).
nearbyint	Computes the element-wise nearbyint of x.
nextafter	Computes the element-wise nextafter of x and y.
pow	Computes arg0 raised to the power of arg1.
reciprocal	Computes the element-wise reciprocal (1/x) of the input tensor.
relu	Computes the element-wise ReLU activation: max(0, x).
rint	Rounds the input tensor to the nearest integer using round-to-nearest-even.
round	Rounds the input tensor to the nearest integer.
signbit	Returns the sign bit of the input tensor.
sinh	Computes the element-wise sinh of x.
tan	Computes the element-wise tangent of the input tensor.
tanh	Computes the element-wise hyperbolic tangent of the input tensor.
trunc	Computes the element-wise trunc of x.