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

Returns:

fpass – The result pass

Return type:

tvm.transform.Pass

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

Simplify

Returns:

fpass – The result pass

Return type:

tvm.transform.Pass

Parameters:

simplify_arguments (bool)

tilelang.transform.simplify.simplify_prim_func(func)

Parameters:

func (Callable)

Return type:

Callable

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

Apply Simplify pass to a PrimFunc or IRModule.

Parameters:

• stmt (Union[tvm.tir.PrimFunc, tvm.IRModule])

• inline_let (bool)

Return type:

Union[tvm.tir.PrimFunc, tvm.IRModule]