did some more work

2020-03-26 21:35:30 +01:00
parent b81938bdb7
commit 78ec3aa4c9
10 changed files with 145 additions and 33 deletions
--- a/thesis/extra_benchmark/benchmark.py
+++ b/thesis/extra_benchmark/benchmark.py
@@ -0,0 +1,54 @@
+import timeit
+
+import numpy as np
+
+from pyqcs.gates.gate import BuiltinGate
+
+cl_state = np.zeros(10, dtype=np.int8)
+qm_state = np.zeros(2**10, dtype=np.cdouble)
+qm_state[0] = 1
+
+gate_uf = BuiltinGate('X', 0, 0, 0.0)
+
+print("running benchmarks ...", end="", flush=True)
+
+time_uf = timeit.repeat("result_uf = gate_uf(qm_state, cl_state)"
+            , setup="import numpy as np;"
+                    "from pyqcs.gates.gate import BuiltinGate;"
+                    "cl_state = np.zeros(10, dtype=np.int8);"
+                    "qm_state = np.zeros(2**10, dtype=np.cdouble);"
+                    "qm_state[0] = 1;"
+                    "gate_uf = BuiltinGate('X', 0, 0, 0.0);"
+            , repeat=5
+            , number=1_000_000
+            )
+
+time_py = timeit.repeat("result_py = np.zeros(2**10, dtype=np.cdouble);"
+                    "result_py[0::2] = qm_state[1::2];"
+                    "result_py[1::2] = qm_state[0::2];"
+                    "cl_py = np.zeros(10, dtype=np.int8)"
+            , setup="import numpy as np;"
+                    "qm_state = np.zeros(2**10, dtype=np.cdouble);"
+                    "qm_state[0] = 1;"
+            , repeat=5
+            , number=1_000_000
+            )
+print(" done")
+
+print("running test ...", end="", flush=True)
+result_uf, cl, m = gate_uf(qm_state, cl_state);
+
+result_py = np.zeros(2**10, dtype=np.cdouble)
+cl_py = np.zeros(10, dtype=np.int8)
+result_py[0::2] = qm_state[1::2]
+result_py[1::2] = qm_state[0::2]
+
+assert np.allclose(result_py, result_uf)
+print(" done")
+
+
+best_uf = min(time_uf) / 1_000_000
+best_py = min(time_py) / 1_000_000
+
+print("time for ufunc gate (best out of 5, 1_000_000 runs):", best_uf)
+print("time for python gate (best out of 5, 1_000_000 runs):", best_py)