tilelang.contrib.cutedsl.math¶

Functions¶

exp10(x[, fastmath])

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

fabsf(x[, fastmath])

divf(x, y[, fastmath])

tanh(x[, fastmath])

Module Contents¶

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.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]