From a896b4683f6fd08a6718e8350e8f4cfc72df4bec Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner Date: Tue, 22 Mar 2022 20:23:05 -0500 Subject: [PATCH] Sort inames before building einsum kernel --- loopy/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/loopy/__init__.py b/loopy/__init__.py index 907d6dfb7..a279b268c 100644 --- a/loopy/__init__.py +++ b/loopy/__init__.py @@ -554,7 +554,7 @@ def make_einsum(spec, arg_names, **knl_creation_kwargs): rhs = rhs * var(arg_name)[tuple(var(i) for i in argsp)] if sum_indices: - rhs = Reduction("sum", tuple(var(idx) for idx in sum_indices), rhs) + rhs = Reduction("sum", tuple(var(idx) for idx in sorted(sum_indices)), rhs) constraints = " and ".join( "0 <= %s < N%s" % (idx, idx) -- GitLab