def transform(self, tree, program_cfg):
arg_cfg, tune_cfg = program_cfg
# tree = Desugar().visit(tree)
inliner = InlineEnvironment(self.symbol_table)
tree = inliner.visit(tree)
tree = PyBasicConversions().visit(tree)
tree.body = inliner.files + tree.body
# tree.find(C.For).pragma = 'omp parallel for'
tree.name = self.original_tree.body[0].name
tree.body.insert(0, StringTemplate("#include <math.h>"))
# print(tree)
return [tree]
def transform(self, tree, program_cfg):
arg_cfg, tune_cfg = program_cfg
# tree = Desugar().visit(tree)
inliner = InlineEnvironment(self.symbol_table)
tree = inliner.visit(tree)
tree = PyBasicConversions().visit(tree)
tree.body = inliner.files + tree.body
# tree.find(C.For).pragma = 'omp parallel for'
tree.name = self.original_tree.body[0].name
tree.body.insert(0, StringTemplate("#include <math.h>"))
# print(tree)
return [tree]
