diff --git a/loopy/symbolic.py b/loopy/symbolic.py index 3a462316654fa947026b5f177b96815b5b05ffab..44be6f9c9e12b51a8adaa51aa1f03bc099d64365 100644 --- a/loopy/symbolic.py +++ b/loopy/symbolic.py @@ -998,7 +998,6 @@ class FunctionToPrimitiveMapper(IdentityMapper): elif name in ["reduce", "simul_reduce"]: - if len(expr.parameters) >= 3: operation, inames = expr.parameters[:2] red_exprs = expr.parameters[2:] diff --git a/loopy/type_inference.py b/loopy/type_inference.py index 34d3fc5e24f37bb2fb9410023398fc1094069090..b8b0cbcbf1236cdf712da998922ac238261a3e6e 100644 --- a/loopy/type_inference.py +++ b/loopy/type_inference.py @@ -637,7 +637,8 @@ def infer_arg_and_reduction_dtypes_for_reduction_expression( import loopy as lp if expr.is_tuple_typed: - arg_dtypes_result = type_inf_mapper(expr, return_tuple=True, return_dtype_set=True) + arg_dtypes_result = type_inf_mapper( + expr, return_tuple=True, return_dtype_set=True) if len(arg_dtypes_result) == 1: arg_dtypes = arg_dtypes_result[0]