diff --git a/loopy/kernel/tools.py b/loopy/kernel/tools.py index 1f7feb1fa23d487f2d0a2be2338b92409e64bfe7..27c0a2218286b57f4756d2f92f106a82c0694135 100644 --- a/loopy/kernel/tools.py +++ b/loopy/kernel/tools.py @@ -106,6 +106,7 @@ def add_and_infer_dtypes(knl, dtype_dict): from loopy.preprocess import infer_unknown_types return infer_unknown_types(knl, expect_completion=True) + def _add_and_infer_dtypes_overdetermined(knl, dtype_dict): knl = _add_dtypes_overdetermined(knl, dtype_dict) @@ -843,7 +844,8 @@ def assign_automatic_axes(kernel, axis=0, local_size=None): # assign longest auto axis inames first auto_axis_inames.sort( - key=lambda iname: (kernel.get_constant_iname_length(iname), iname), + key=lambda iname: ( + kernel.get_constant_iname_length(iname), iname), reverse=True) if auto_axis_inames: