From c700eb6de3f2f082a33b329d72086eee1ce3b44a Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Fri, 11 Nov 2011 20:04:05 -0500
Subject: [PATCH] Switch ill-advised outer 'unroll' on split automatic axis
back to sequential.
---
loopy/preprocess.py | 6 +++++-
1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/loopy/preprocess.py b/loopy/preprocess.py
index 0a374edf7..bef261ba1 100644
--- a/loopy/preprocess.py
+++ b/loopy/preprocess.py
@@ -524,9 +524,13 @@ def assign_automatic_axes(kernel, axis=0, local_size=None):
new_tag = LocalIndexTag(axis)
if desired_length > local_size[axis]:
from loopy import split_dimension
+
+ # Don't be tempted to switch the outer tag to unroll--this may
+ # generate tons of code on some examples.
+
return assign_automatic_axes(
split_dimension(kernel, iname, inner_length=local_size[axis],
- outer_tag=UnrollTag(), inner_tag=new_tag,
+ outer_tag=None, inner_tag=new_tag,
do_tagged_check=False),
axis=recursion_axis, local_size=local_size)
--
GitLab