transforms.artiq_ir_generator: add tests for devirtualization.

m-labs · Oct 8, 2015 · b6c8c9f · b6c8c9f
1 parent 0bb7931
commit b6c8c9f
Showing 2 changed files with 30 additions and 0 deletions.
diff --git a/lit-test/test/devirtualization/ddb.pyon b/lit-test/test/devirtualization/ddb.pyon
@@ -0,0 +1,8 @@
+{
+    "comm": {
+        "type": "local",
+        "module": "artiq.coredevice.comm_dummy",
+        "class": "Comm",
+        "arguments": {}
+    }
+}
diff --git a/lit-test/test/devirtualization/function.py b/lit-test/test/devirtualization/function.py
@@ -0,0 +1,22 @@
+# RUN: env ARTIQ_DUMP_IR=1 %python -m artiq.compiler.testbench.embedding +compile %s 2>%t
+# RUN: OutputCheck %s --file-to-check=%t
+
+from artiq.language.core import *
+from artiq.language.types import *
+
+# CHECK-L: call ()->NoneType %local.testbench.entrypoint ; calls testbench.entrypoint
+
+@kernel
+def baz():
+    pass
+
+class foo:
+    @kernel
+    def bar(self):
+        # CHECK-L: call ()->NoneType %local.testbench.baz ; calls testbench.baz
+        baz()
+x = foo()
+
+@kernel
+def entrypoint():
+    x.bar()