tilelang.transform.simplify

Functions

Simplify([simplify_arguments])	Simplify
simplify_prim_func(func)
apply_simplify(stmt)	Apply Simplify pass to a PrimFunc or IRModule.

Module Contents

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)

Apply Simplify pass to a PrimFunc or IRModule.

Parameters:

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

Return type:

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