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

参数:

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

• fastmath (bool)

返回类型:

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

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

参数:

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

• fastmath (bool)

返回类型:

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

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

参数:

• 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)

返回类型:

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

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

参数:

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

• fastmath (bool)

返回类型:

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