final first commit

qibojit-benchmarks/plots/devices.py

@@ -0,0 +1,44 @@
+"""Scaling plots with performance comparison of qibojit backend run on different CPU and GPU devices."""
+import matplotlib
+import matplotlib.pyplot as plt
+import seaborn as sns
+matplotlib.rcParams['mathtext.fontset'] = 'cm'
+matplotlib.rcParams['font.family'] = 'STIXGeneral'
+
+
+class Line:
+
+    def __init__(self, label, data, color, marker, linestyle="-"):
+        self.label = label
+        self.data = data
+        self.color = color
+        self.marker = marker
+        self.linestyle = linestyle
+
+
+def plot_devices(lines, circuit, quantity, precision="double", 
+                 fontsize=30, legendfont=None, save=False):
+    matplotlib.rcParams["font.size"] = fontsize
+    # Filter data
+    for line in lines:
+        condition = (line.data["circuit"] == circuit) & (line.data["precision"] == precision)
+        line.data = line.data[condition]
+
+    plt.figure(figsize=(14, 8))
+    for line in lines:
+        plt.semilogy(line.data["nqubits"], line.data[quantity], color=line.color, linestyle=line.linestyle,
+                     linewidth=3.0, marker=line.marker, markersize=10, label=line.label)
+
+    plt.title(f"qibojit, {circuit}, {precision} precision")
+    plt.xlabel("Number of qubits")
+    if quantity == "total_dry_time":
+        plt.ylabel("Total dry run time (sec)")
+    elif quantity == "total_simulation_time":
+        plt.ylabel("Total simulation time (sec)")
+
+    if legendfont is not None:
+        plt.legend(loc="upper left", fontsize=legendfont)
+    if save:
+        plt.savefig(f"devices_{circuit}_{quantity}_{precision}.pdf", bbox_inches="tight")
+    else:
+        plt.show()