forked from M-Labs/artiq
Add ARTIQ_DUMP_ASSEMBLY.
This commit is contained in:
parent
9c5ca2ae29
commit
8b7d38d203
|
@ -1,4 +1,4 @@
|
|||
import tempfile, subprocess
|
||||
import os, sys, tempfile, subprocess
|
||||
from llvmlite_artiq import ir as ll, binding as llvm
|
||||
|
||||
llvm.initialize()
|
||||
|
@ -56,10 +56,20 @@ class Target:
|
|||
|
||||
def compile(self, module):
|
||||
"""Compile the module to a relocatable object for this target."""
|
||||
|
||||
if os.getenv('ARTIQ_DUMP_IR'):
|
||||
print("====== ARTIQ IR DUMP ======", file=sys.stderr)
|
||||
for function in module.artiq_ir:
|
||||
print(function, file=sys.stderr)
|
||||
|
||||
llmod = module.build_llvm_ir(self)
|
||||
llparsedmod = llvm.parse_assembly(str(llmod))
|
||||
llparsedmod.verify()
|
||||
|
||||
if os.getenv('ARTIQ_DUMP_LLVM'):
|
||||
print("====== LLVM IR DUMP ======", file=sys.stderr)
|
||||
print(str(llparsedmod), file=sys.stderr)
|
||||
|
||||
llpassmgrbuilder = llvm.create_pass_manager_builder()
|
||||
llpassmgrbuilder.opt_level = 2 # -O2
|
||||
llpassmgrbuilder.size_level = 1 # -Os
|
||||
|
@ -68,10 +78,19 @@ class Target:
|
|||
llpassmgrbuilder.populate(llpassmgr)
|
||||
llpassmgr.run(llparsedmod)
|
||||
|
||||
if os.getenv('ARTIQ_DUMP_LLVM'):
|
||||
print("====== LLVM IR DUMP (OPTIMIZED) ======", file=sys.stderr)
|
||||
print(str(llparsedmod), file=sys.stderr)
|
||||
|
||||
lltarget = llvm.Target.from_triple(self.triple)
|
||||
llmachine = lltarget.create_target_machine(
|
||||
features=",".join(self.features),
|
||||
reloc="pic", codemodel="default")
|
||||
|
||||
if os.getenv('ARTIQ_DUMP_ASSEMBLY'):
|
||||
print("====== ASSEMBLY DUMP ======", file=sys.stderr)
|
||||
print(llmachine.emit_assembly(llparsedmod), file=sys.stderr)
|
||||
|
||||
return llmachine.emit_object(llparsedmod)
|
||||
|
||||
def compile_and_link(self, modules):
|
||||
|
|
|
@ -1,4 +1,4 @@
|
|||
import os, sys, tempfile
|
||||
import sys, tempfile
|
||||
|
||||
from pythonparser import diagnostic
|
||||
|
||||
|
@ -35,15 +35,6 @@ class Core:
|
|||
module = Module(stitcher)
|
||||
target = OR1KTarget()
|
||||
|
||||
if os.getenv('ARTIQ_DUMP_IR'):
|
||||
print("====== ARTIQ IR DUMP ======", file=sys.stderr)
|
||||
for function in module.artiq_ir:
|
||||
print(function, file=sys.stderr)
|
||||
|
||||
if os.getenv('ARTIQ_DUMP_LLVM'):
|
||||
print("====== LLVM IR DUMP ======", file=sys.stderr)
|
||||
print(module.build_llvm_ir(target), file=sys.stderr)
|
||||
|
||||
return target.compile_and_link([module]), stitcher.rpc_map
|
||||
except diagnostic.Error as error:
|
||||
print("\n".join(error.diagnostic.render(colored=True)), file=sys.stderr)
|
||||
|
|
Loading…
Reference in New Issue