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 #