diff --git a/loopy/transform/padding.py b/loopy/transform/padding.py index 9afc28f9fb8dae52b0f4a0f5f7e14ae368784b56..7771185e0af3b02ebb8697f9319515108a3cf0a9 100644 --- a/loopy/transform/padding.py +++ b/loopy/transform/padding.py @@ -379,7 +379,7 @@ def _split_array_axis_inner(kernel, array_name, axis_nr, count, order="C"): else: raise RuntimeError("order '%s' not understood" % order) - return expr.aggregate.index(tuple(idx)) + return expr.aggregate[tuple(idx)] rule_mapping_context = SubstitutionRuleMappingContext( kernel.substitutions, var_name_gen)