From 6b44618970b04a5816660a7684a8d5f98a13b290 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Thu, 13 Jun 2013 23:07:28 -0400
Subject: [PATCH] Let DomainParameterFinder find *any* integer argument
---
loopy/compiled.py | 4 +---
1 file changed, 1 insertion(+), 3 deletions(-)
diff --git a/loopy/compiled.py b/loopy/compiled.py
index c1a977c57..c7535ccad 100644
--- a/loopy/compiled.py
+++ b/loopy/compiled.py
@@ -143,8 +143,6 @@ class DomainParameterFinder(object):
# returning the desired parameter.
self.param_to_sources = param_to_sources = {}
- param_names = kernel.all_params()
-
from loopy.kernel.data import GlobalArg
from loopy.symbolic import DependencyMapper
from pymbolic import compile
@@ -158,7 +156,7 @@ class DomainParameterFinder(object):
if len(deps) == 1:
dep, = deps
- if dep.name in param_names:
+ if kernel.arg_dict[dep.name].dtype.kind == "i":
from pymbolic.algorithm import solve_affine_equations_for
try:
# friggin' overkill :)
--
GitLab