From be5f165ff6485828b7a8cb0349a05032453785fb Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner Date: Wed, 2 Nov 2011 13:11:00 -0400 Subject: [PATCH] Add a force flag to tag_dimensions. --- loopy/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/loopy/__init__.py b/loopy/__init__.py index 451a17583..ba05ddff4 100644 --- a/loopy/__init__.py +++ b/loopy/__init__.py @@ -381,7 +381,7 @@ def join_dimensions(kernel, inames, new_iname=None, tag=AutoFitLocalIndexTag()): # {{{ dimension tag -def tag_dimensions(kernel, iname_to_tag): +def tag_dimensions(kernel, iname_to_tag, force=False): from loopy.kernel import parse_tag iname_to_tag = dict((iname, parse_tag(tag)) @@ -406,7 +406,7 @@ def tag_dimensions(kernel, iname_to_tag): raise ValueError("cannot tag '%s' as parallel--" "iname requires sequential execution" % iname) - if old_tag is not None and (old_tag != new_tag): + if (not force) and old_tag is not None and (old_tag != new_tag): raise RuntimeError("'%s' is already tagged '%s'--cannot retag" % (iname, old_tag)) -- GitLab