4.9. Examples — Anaconda Enterprise 4.4 documentation

Anaconda Enterprise: Documentation

# 4.9. Examples

## 4.9.1. Compiling a trivial function

Compile and execute the function defined in A trivial function. The function is compiled with no specific optimizations.

    from __future__ import print_function

    from ctypes import CFUNCTYPE, c_double

    import llvmlite.binding as llvm

    # All these initializations are required for code generation!
    llvm.initialize()
    llvm.initialize_native_target()
    llvm.initialize_native_asmprinter()  # yes, even this one

    llvm_ir = """
       ; ModuleID = "examples/ir_fpadd.py"
       target triple = "unknown-unknown-unknown"
       target datalayout = ""

       define double @"fpadd"(double %".1", double %".2")
       {
       entry:
         %"res" = fadd double %".1", %".2"
         ret double %"res"
       }
       """

    def create_execution_engine():
        """
        Create an ExecutionEngine suitable for JIT code generation on
        the host CPU.  The engine is reusable for an arbitrary number of
        modules.
        """
        # Create a target machine representing the host
        target = llvm.Target.from_default_triple()
        target_machine = target.create_target_machine()
        # And an execution engine with an empty backing module
        backing_mod = llvm.parse_assembly("")
        engine = llvm.create_mcjit_compiler(backing_mod, target_machine)
        return engine

    def compile_ir(engine, llvm_ir):
        """
        Compile the LLVM IR string with the given engine.
        The compiled module object is returned.
        """
        # Create a LLVM module object from the IR
        mod = llvm.parse_assembly(llvm_ir)
        mod.verify()
        # Now add the module and make sure it is ready for execution
        engine.add_module(mod)
        engine.finalize_object()
        return mod

    engine = create_execution_engine()
    mod = compile_ir(engine, llvm_ir)

    # Look up the function pointer (a Python int)
    func_ptr = engine.get_function_address("fpadd")

    # Run the function via ctypes
    cfunc = CFUNCTYPE(c_double, c_double, c_double)(func_ptr)
    res = cfunc(1.0, 3.5)
    print("fpadd(...) =", res)