def map_variable(self, expr): match_obj = INDEXED_VAR_RE.match(expr.name) if match_obj is not None: name = match_obj.group(1) subscript = int(match_obj.group(2)) if name in self.name_whitelist: return prim.Variable(name).index(subscript) else: return IdentityMapper.map_variable(self, expr) else: return IdentityMapper.map_variable(self, expr)
def map_variable(self, expr): match_obj = INDEXED_VAR_RE.match(expr.name) if match_obj is not None: name = match_obj.group(1) subscript = int(match_obj.group(2)) if name in self.name_whitelist: return prim.Variable(name).index(subscript) else: return IdentityMapper.map_variable(self, expr) else: return IdentityMapper.map_variable(self, expr)
def map_variable(self, expr): if expr.name == "pi": return prim.Variable("M_PI") else: return IdentityMapper.map_variable(self, expr)
def map_variable(self, expr): if expr.name == "pi": return prim.Variable("M_PI") else: return IdentityMapper.map_variable(self, expr)