Return LLVM IR module from LLVMIRGenerator.process.

whitequark · whitequark · commit dff4ce7e3adb · 2015-07-22T04:13:04.000+03:00
diff --git a/artiq/compiler/module.py b/artiq/compiler/module.py
@@ -34,8 +34,7 @@ def __init__(self, source_buffer, engine=None):
         self.artiq_ir = artiq_ir_generator.visit(self.typedtree)
         dead_code_eliminator.process(self.artiq_ir)
         local_access_validator.process(self.artiq_ir)
-        llvm_ir_generator.process(self.artiq_ir)
-        self.llvm_ir = llvm_ir_generator.llmodule
+        self.llvm_ir = llvm_ir_generator.process(self.artiq_ir)
 
     @classmethod
     def from_string(cls, source_string, name="input.py", first_line=1, engine=None):
diff --git a/artiq/compiler/transforms/llvm_ir_generator.py b/artiq/compiler/transforms/llvm_ir_generator.py
@@ -130,6 +130,8 @@ def process(self, functions):
         for func in functions:
             self.process_function(func)
 
+        return self.llmodule
+
     def process_function(self, func):
         try:
             self.llfunction = self.llmodule.get_global(func.name)
