diff --git a/loopy/padding.py b/loopy/padding.py index 7998dd4ba80370c500c6a853e181a4b33332f778..65d90a8f48c86579e0d444306e8de3e6d10dfa85 100644 --- a/loopy/padding.py +++ b/loopy/padding.py @@ -23,13 +23,9 @@ THE SOFTWARE. """ - - from loopy.symbolic import ExpandingIdentityMapper - - class ArgAxisSplitHelper(ExpandingIdentityMapper): def __init__(self, rules, var_name_gen, arg_names, handler): ExpandingIdentityMapper.__init__(self, rules, var_name_gen) @@ -43,9 +39,6 @@ class ArgAxisSplitHelper(ExpandingIdentityMapper): return ExpandingIdentityMapper.map_subscript(self, expr, expn_state) - - - def split_arg_axis(kernel, args_and_axes, count): """ :arg args_and_axes: a list of tuples *(arg, axis_nr)* indicating @@ -191,8 +184,6 @@ def split_arg_axis(kernel, args_and_axes, count): return kernel - - def find_padding_multiple(kernel, variable, axis, align_bytes, allowed_waste=0.1): arg = kernel.arg_dict[variable] @@ -214,8 +205,6 @@ def find_padding_multiple(kernel, variable, axis, align_bytes, allowed_waste=0.1 multiple += 1 - - def add_padding(kernel, variable, axis, align_bytes): arg_to_idx = dict((arg.name, i) for i, arg in enumerate(kernel.args)) arg_idx = arg_to_idx[variable]