diff --git a/loopy/transform/data.py b/loopy/transform/data.py index 5b1ee6ccafa4c3b76609f197cc31691de562aaa5..7fbc595f27ea34f9056b36e3d6f0e168b182c24e 100644 --- a/loopy/transform/data.py +++ b/loopy/transform/data.py @@ -387,6 +387,15 @@ def change_arg_to_image(knl, name): def tag_array_axes(knl, ary_names, dim_tags): """ + :arg dim_tags: a tuple of + :class:`loopy.kernel.array.ArrayDimImplementationTag` or a string that + parses to one. See :func:`loopy.kernel.array.parse_dim_tags` for a + description of the allowed string format. + + For example, *dim_tags* could be ``"N2,N0,N1"`` to determine + that the second axis is the fastest-varying, the last is + the next-fastest, and the first is the slowest. + .. versionchanged:: 2016.2 This function was called :func:`tag_data_axes` before version 2016.2.