mirror of https://github.com/m-labs/artiq
36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
import os, sys, fileinput, ctypes
|
|
from pythonparser import diagnostic
|
|
from llvmlite_artiq import binding as llvm
|
|
from ..module import Module, Source
|
|
from ..targets import NativeTarget
|
|
|
|
def main():
|
|
libartiq_support = os.getenv("LIBARTIQ_SUPPORT")
|
|
if libartiq_support is not None:
|
|
llvm.load_library_permanently(libartiq_support)
|
|
|
|
def process_diagnostic(diag):
|
|
print("\n".join(diag.render()))
|
|
if diag.level in ("fatal", "error"):
|
|
exit(1)
|
|
|
|
engine = diagnostic.Engine()
|
|
engine.process = process_diagnostic
|
|
|
|
source = "".join(fileinput.input())
|
|
source = source.replace("#ARTIQ#", "")
|
|
mod = Module(Source.from_string(source.expandtabs(), engine=engine))
|
|
|
|
target = NativeTarget()
|
|
llmod = mod.build_llvm_ir(target)
|
|
llparsedmod = llvm.parse_assembly(str(llmod))
|
|
llparsedmod.verify()
|
|
|
|
llmachine = llvm.Target.from_triple(target.triple).create_target_machine()
|
|
lljit = llvm.create_mcjit_compiler(llparsedmod, llmachine)
|
|
llmain = lljit.get_pointer_to_global(llparsedmod.get_function(llmod.name + ".__modinit__"))
|
|
ctypes.CFUNCTYPE(None)(llmain)()
|
|
|
|
if __name__ == "__main__":
|
|
main()
|