Return LLVM IR module from LLVMIRGenerator.process.

artiq/compiler/module.py

@@ -34,8 +34,7 @@ class Module:
         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):

artiq/compiler/transforms/llvm_ir_generator.py

@@ -130,6 +130,8 @@ class LLVMIRGenerator:
         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)