diff --git a/loopy/kernel.py b/loopy/kernel.py index f35e936ae0d76c1c9a6c849ad396fff605547812..034e7b05541ae54e02a46223a118c8fd741b7daa 100644 --- a/loopy/kernel.py +++ b/loopy/kernel.py @@ -872,6 +872,19 @@ class LoopKernel(Record): if var_name not in used_vars: return var_name + def get_var_descriptor(self, name): + try: + return self.arg_dict[name] + except KeyError: + pass + + try: + return self.temporary_variables[name] + except KeyError: + pass + + raise ValueError("nothing known about variable '%s'" % name) + @property @memoize_method def id_to_insn(self):