From 94b1b76e715c6207f65cc261cb5fa432d645c4b3 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Mon, 2 Apr 2012 01:24:14 -0400
Subject: [PATCH] Add LoopKernel.get_var_descriptor().
---
loopy/kernel.py | 13 +++++++++++++
1 file changed, 13 insertions(+)
diff --git a/loopy/kernel.py b/loopy/kernel.py
index f35e936ae..034e7b055 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):
--
GitLab