Add full depolarising example

xdslproject · Oct 29, 2024 · b2945da · b2945da
1 parent eb0c03d
commit b2945da
Show file tree

Hide file tree

Showing 2 changed files with 41 additions and 0 deletions.
diff --git a/inconspiquous/dialects/__init__.py b/inconspiquous/dialects/__init__.py
@@ -76,6 +76,11 @@ def get_test():
 
         return Test
 
+    def get_varith():
+        from xdsl.dialects.varith import Varith
+
+        return Varith
+
     return {
         "arith": get_arith,
         "builtin": get_builtin,
@@ -91,4 +96,5 @@ def get_test():
         "stim": get_stim,
         "tensor": get_tensor,
         "test": get_test,
+        "varith": get_varith,
     }
diff --git a/tests/filecheck/examples/depolarising.mlir b/tests/filecheck/examples/depolarising.mlir
@@ -1,5 +1,40 @@
 // RUN: QUOPT_ROUNDTRIP
 
+// CHECK:      func.func @depolarising_dyn(%q : !qubit.bit) -> !qubit.bit {
+// CHECK-NEXT:   %p = prob.bernoulli 1.000000e-01 : f64
+// CHECK-NEXT:   %id = gate.constant #gate.id
+// CHECK-NEXT:   %p2 = prob.uniform : i2
+// CHECK-NEXT:   %x = gate.constant #gate.x
+// CHECK-NEXT:   %y = gate.constant #gate.y
+// CHECK-NEXT:   %z = gate.constant #gate.z
+// CHECK-NEXT:   %choice = varith.switch %p2 : i2 -> !gate.type<1>, [
+// CHECK-NEXT:     default: %id,
+// CHECK-NEXT:     1: %x,
+// CHECK-NEXT:     2: %y,
+// CHECK-NEXT:     3: %z
+// CHECK-NEXT:   ]
+// CHECK-NEXT:   %g = arith.select %p, %choice, %id : !gate.type<1>
+// CHECK-NEXT:   %q1 = qssa.dyn_gate<%g> %q : !qubit.bit
+// CHECK-NEXT:   func.return %q1 : !qubit.bit
+// CHECK-NEXT: }
+func.func @depolarising_dyn(%q : !qubit.bit) -> !qubit.bit {
+  %p = prob.bernoulli 0.1
+  %id = gate.constant #gate.id
+  %p2 = prob.uniform : i2
+  %x = gate.constant #gate.x
+  %y = gate.constant #gate.y
+  %z = gate.constant #gate.z
+  %choice = varith.switch %p2 : i2 -> !gate.type<1>, [
+    default: %id,
+    1: %x,
+    2: %y,
+    3: %z
+  ]
+  %g = arith.select %p, %choice, %id : !gate.type<1>
+  %q1 = qssa.dyn_gate<%g> %q : !qubit.bit
+  func.return %q1 : !qubit.bit
+}
+
 // CHECK:      func.func @depolarising_scf(%q : !qubit.bit) -> !qubit.bit {
 // CHECK-NEXT:   %p = prob.bernoulli 1.000000e-01 : f64
 // CHECK-NEXT:   %q3 = scf.if %p -> (!qubit.bit) {