From ca06c62d16a1260d97195956038cec18da7ace0f Mon Sep 17 00:00:00 2001 From: Dan Lykov Date: Thu, 8 Feb 2024 00:13:03 -0600 Subject: [PATCH] expand data from the benchmarking --- qtensor/tools/benchmarking/simulators.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/qtensor/tools/benchmarking/simulators.py b/qtensor/tools/benchmarking/simulators.py index 659bc775..cebc4861 100644 --- a/qtensor/tools/benchmarking/simulators.py +++ b/qtensor/tools/benchmarking/simulators.py @@ -162,12 +162,13 @@ def optimize_qaoa_energy(self, G, p, opt_time = 0 ests = [] opts = [] + times = [] for edge in self.iterate_edges(G, p): with profiles.timing() as t: circuit = sim._edge_energy_circuit(G, gamma, beta, edge) tn = qtensor.optimisation.TensorNet.QtreeTensorNet.from_qtree_gates(circuit) peo, _ = opt.optimize(tn) - opt_time += t.result + times.append(t.result) mems, flops = tn.simulation_cost(peo) ests.append(ContractionEstimation( width=opt.treewidth, @@ -175,7 +176,7 @@ def optimize_qaoa_energy(self, G, p, flops=sum(flops) )) opts.append(peo) - return opts, ests, opt_time + return opts, ests, times def simulate_qaoa_energy(self, G, p, opt): gamma, beta = get_test_gamma_beta(p) @@ -437,7 +438,7 @@ def optimize_qaoa_energy(self, G, p, opt_type=None, infos.append((rehs['info'], rehs['tn'])) ests.append(self._rehs2est(rehs)) - return infos, ests, sum(times) + return infos, ests, times def simulate_qaoa_energy(self, G, p, opts,