From 7690a127347830001dbbfb23f0224dccd87d7480 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Wed, 2 Nov 2011 13:13:32 -0400
Subject: [PATCH] Add a switch for annotation in the CCodeMapper.
---
loopy/symbolic.py | 11 ++++++++---
1 file changed, 8 insertions(+), 3 deletions(-)
diff --git a/loopy/symbolic.py b/loopy/symbolic.py
index 83fd8bde3..a5bc8ea12 100644
--- a/loopy/symbolic.py
+++ b/loopy/symbolic.py
@@ -258,7 +258,7 @@ class ArrayAccessFinder(CombineMapper):
# {{{ C code mapper
class LoopyCCodeMapper(CCodeMapper):
- def __init__(self, kernel, cse_name_list=[], var_subst_map={}):
+ def __init__(self, kernel, cse_name_list=[], var_subst_map={}, with_annotation=True):
def constant_mapper(c):
if isinstance(c, float):
# FIXME: type-variable
@@ -270,6 +270,7 @@ class LoopyCCodeMapper(CCodeMapper):
cse_name_list=cse_name_list)
self.kernel = kernel
+ self.with_annotation = with_annotation
self.var_subst_map = var_subst_map.copy()
def copy(self, var_subst_map=None, cse_name_list=None):
@@ -292,8 +293,12 @@ class LoopyCCodeMapper(CCodeMapper):
def map_variable(self, expr, prec):
if expr.name in self.var_subst_map:
- return " /* %s */ %s" % (
- expr.name, self.rec(self.var_subst_map[expr.name], prec))
+ if self.with_annotation:
+ return " /* %s */ %s" % (
+ expr.name,
+ self.rec(self.var_subst_map[expr.name], prec))
+ else:
+ return str(self.rec(self.var_subst_map[expr.name], prec))
else:
return CCodeMapper.map_variable(self, expr, prec)
--
GitLab