transforms.interleaver: add boilerplate.

m-labs · Nov 17, 2015 · 48a2bb1 · 48a2bb1
1 parent de9d7eb
commit 48a2bb1
Showing 3 changed files with 25 additions and 0 deletions.
diff --git a/artiq/compiler/module.py b/artiq/compiler/module.py
@@ -58,6 +58,7 @@ def __init__(self, src, ref_period=1e-6):
         dead_code_eliminator = transforms.DeadCodeEliminator(engine=self.engine)
         local_access_validator = validators.LocalAccessValidator(engine=self.engine)
         devirtualization = analyses.Devirtualization()
+        interleaver = transforms.Interleaver(engine=self.engine)
 
         self.name = src.name
         self.globals = src.globals
@@ -71,6 +72,7 @@ def __init__(self, src, ref_period=1e-6):
         artiq_ir_generator.annotate_calls(devirtualization)
         dead_code_eliminator.process(self.artiq_ir)
         local_access_validator.process(self.artiq_ir)
+        interleaver.process(self.artiq_ir)
 
     def build_llvm_ir(self, target):
         """Compile the module to LLVM IR for the specified target."""

diff --git a/artiq/compiler/transforms/__init__.py b/artiq/compiler/transforms/__init__.py
@@ -5,3 +5,4 @@
 from .artiq_ir_generator import ARTIQIRGenerator
 from .dead_code_eliminator import DeadCodeEliminator
 from .llvm_ir_generator import LLVMIRGenerator
+from .interleaver import Interleaver
diff --git a/artiq/compiler/transforms/interleaver.py b/artiq/compiler/transforms/interleaver.py
@@ -0,0 +1,22 @@
+"""
+:class:`Interleaver` reorders requests to the RTIO core so that
+the timestamp would always monotonically nondecrease.
+"""
+
+from .. import ir
+from ..analyses import domination
+
+class Interleaver:
+    def __init__(self, engine):
+        self.engine = engine
+
+    def process(self, functions):
+        for func in functions:
+            self.process_function(func)
+
+    def process_function(self, func):
+        domtree = domination.PostDominatorTree(func)
+        print(func)
+        for block in func.basic_blocks:
+            idom = domtree.immediate_dominator(block)
+            print(block.name, "->", idom.name if idom else "<exit>")