tilelang.transform.simplify

Functions

LetInline()

LetInline

Simplify([simplify_arguments])

Simplify

simplify_prim_func(func)

apply_simplify(stmt[, inline_let])

Apply Simplify pass to a PrimFunc or IRModule.

Module Contents

tilelang.transform.simplify.LetInline()

LetInline

返回:

fpass -- The result pass

返回类型:

tvm.transform.Pass

tilelang.transform.simplify.Simplify(simplify_arguments=False)

Simplify

返回:

fpass -- The result pass

返回类型:

tvm.transform.Pass

参数:

simplify_arguments (bool)

tilelang.transform.simplify.simplify_prim_func(func)
参数:

func (Callable)

返回类型:

Callable

tilelang.transform.simplify.apply_simplify(stmt, inline_let=False)

Apply Simplify pass to a PrimFunc or IRModule.

参数:
  • stmt (tvm.tir.PrimFunc | tvm.IRModule)

  • inline_let (bool)

返回类型:

tvm.tir.PrimFunc | tvm.IRModule