From 9d3e11134aa91155a9f564cbf7e2493ea3f27d3e Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Sun, 22 Sep 2019 18:46:39 +0200
Subject: [PATCH] Add target option to LLVMPrinter
---
pystencils/llvm/llvm.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/pystencils/llvm/llvm.py b/pystencils/llvm/llvm.py
index de02fdec..4e730f04 100644
--- a/pystencils/llvm/llvm.py
+++ b/pystencils/llvm/llvm.py
@@ -35,7 +35,7 @@ def generate_llvm(ast_node, module=None, builder=None):
class LLVMPrinter(Printer):
"""Convert expressions to LLVM IR"""
- def __init__(self, module, builder, fn=None, *args, **kwargs):
+ def __init__(self, module, builder, fn=None, target='cpu', *args, **kwargs):
self.func_arg_map = kwargs.pop("func_arg_map", {})
super(LLVMPrinter, self).__init__(*args, **kwargs)
self.fp_type = ir.DoubleType()
@@ -48,6 +48,7 @@ class LLVMPrinter(Printer):
self.fn = fn
self.ext_fn = {} # keep track of wrappers to external functions
self.tmp_var = {}
+ self.target = target
def _add_tmp_var(self, name, value):
self.tmp_var[name] = value
--
GitLab