tilelang.contrib.cutedsl.math¶

Functions¶

exp(x[, fastmath])

exp2(x[, fastmath])

log(x[, fastmath])

log1p(x[, fastmath])

log2(x[, fastmath])

log10(x[, fastmath])

tan(x[, fastmath])

cos(x[, fastmath])

sin(x[, fastmath])

sqrt(x[, fastmath])

rsqrt(x[, fastmath])

exp10(x[, fastmath])

Compute 10^x using exp2(x * log2(10)).

fabsf(x[, fastmath])

abs2(x)

max2(x, y)

min2(x, y)

copysignf(x, y[, fastmath])

isfinite(x)

divf(x, y[, fastmath])

tanh(x[, fastmath])

Module Contents¶

tilelang.contrib.cutedsl.math.exp(x, fastmath=False, **kwargs)¶
Parameters:

  • x (cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric])

  • fastmath (bool)

Return type:

cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric]

tilelang.contrib.cutedsl.math.exp2(x, fastmath=False, **kwargs)¶
Parameters:

  • x (cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric])

  • fastmath (bool)

Return type:

cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric]

tilelang.contrib.cutedsl.math.log(x, fastmath=False, **kwargs)¶
Parameters:

  • x (cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric])

  • fastmath (bool)

Return type:

cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric]

tilelang.contrib.cutedsl.math.log1p(x, fastmath=False, **kwargs)¶
Parameters:

  • x (cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric])

  • fastmath (bool)

Return type:

cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric]

tilelang.contrib.cutedsl.math.log2(x, fastmath=False, **kwargs)¶
Parameters:

  • x (cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric])

  • fastmath (bool)

Return type:

cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric]

tilelang.contrib.cutedsl.math.log10(x, fastmath=False, **kwargs)¶
Parameters:

  • x (cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric])

  • fastmath (bool)

Return type:

cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric]

tilelang.contrib.cutedsl.math.tan(x, fastmath=False, **kwargs)¶
Parameters:

  • x (cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric])

  • fastmath (bool)

Return type:

cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric]

tilelang.contrib.cutedsl.math.cos(x, fastmath=False, **kwargs)¶
Parameters:

  • x (cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric])

  • fastmath (bool)

Return type:

cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric]

tilelang.contrib.cutedsl.math.sin(x, fastmath=False, **kwargs)¶
Parameters:

  • x (cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric])

  • fastmath (bool)

Return type:

cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric]

tilelang.contrib.cutedsl.math.sqrt(x, fastmath=False, **kwargs)¶
Parameters:

  • x (cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric])

  • fastmath (bool)

Return type:

cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric]

tilelang.contrib.cutedsl.math.rsqrt(x, fastmath=False, **kwargs)¶
Parameters:

  • x (cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric])

  • fastmath (bool)

Return type:

cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric]

tilelang.contrib.cutedsl.math.exp10(x, fastmath=False)¶

Compute 10^x using exp2(x * log2(10)).

Parameters:

  • x (cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric])

  • fastmath (bool)

Return type:

cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric]

tilelang.contrib.cutedsl.math.fabsf(x, fastmath=False)¶
Parameters:

  • x (cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric])

  • fastmath (bool)

Return type:

cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric]

tilelang.contrib.cutedsl.math.abs2(x)¶
Parameters:

x (cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric])

Return type:

cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric]

tilelang.contrib.cutedsl.math.max2(x, y)¶
Parameters:

  • x (cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric])

  • y (cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric])

Return type:

cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric]

tilelang.contrib.cutedsl.math.min2(x, y)¶
Parameters:

  • x (cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric])

  • y (cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric])

Return type:

cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric]

tilelang.contrib.cutedsl.math.copysignf(x, y, fastmath=False)¶
Parameters:

  • x (cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric])

  • y (cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric])

  • fastmath (bool)

Return type:

cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric]

tilelang.contrib.cutedsl.math.isfinite(x)¶
Parameters:

x (cutlass.cute.typing.Numeric)

Return type:

cutlass.Boolean

tilelang.contrib.cutedsl.math.divf(x, y, fastmath=False)¶
Parameters:

  • x (cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric])

  • y (cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric])

  • fastmath (bool)

Return type:

cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric]

tilelang.contrib.cutedsl.math.tanh(x, fastmath=False)¶
Parameters:

  • x (cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric])

  • fastmath (bool)

Return type:

cutlass.cute.typing.Union[cutlass.cute.tensor.TensorSSA, cutlass.cute.typing.Numeric]