diff --git a/loopy/compiled.py b/loopy/compiled.py index c1a977c5795c42771eea6023c56ee2007b74c8e5..c7535ccaddaf19024e62a896763676ef31782a21 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 :)