tilelang.transform.simplify module

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

Returns:

fpass – The result pass

Return type:

tvm.transform.Pass

tilelang.transform.simplify.apply_simplify(stmt: Union[PrimFunc, IRModule]) → Union[PrimFunc, IRModule]

Apply Simplify pass to a PrimFunc or IRModule.

tilelang.transform.simplify.simplify_prim_func(func: Callable) → Callable