From 7c510c29d1122aa41d48b3d0e5e975dad2250f0e Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Sat, 28 Nov 2015 15:43:36 -0500
Subject: [PATCH] Precompute: Only assign automatic axes if automatic axis were
created
---
loopy/precompute.py | 10 ++++++++--
1 file changed, 8 insertions(+), 2 deletions(-)
diff --git a/loopy/precompute.py b/loopy/precompute.py
index 18d066397..a8b082d0a 100644
--- a/loopy/precompute.py
+++ b/loopy/precompute.py
@@ -839,8 +839,14 @@ def precompute(kernel, subst_use, sweep_inames=[], within=None,
from loopy import tag_inames
kernel = tag_inames(kernel, new_iname_to_tag)
- from loopy.kernel.tools import assign_automatic_axes
- kernel = assign_automatic_axes(kernel)
+ from loopy.kernel.data import AutoFitLocalIndexTag
+ has_automatic_axes = any(
+ isinstance(tag, AutoFitLocalIndexTag)
+ for tag in new_iname_to_tag.values())
+
+ if has_automatic_axes:
+ from loopy.kernel.tools import assign_automatic_axes
+ kernel = assign_automatic_axes(kernel)
return kernel
--
GitLab