plot scripts updated.

plot_scripts/compare_expectation.py

@@ -0,0 +1,80 @@
+import numpy as np
+import h5py
+# from map_packages_colors import *
+import matplotlib.pyplot as plt
+
+import os
+import sys
+
+import warnings
+warnings.filterwarnings("ignore")
+
+
+params = {'backend': 'ps',
+          'axes.labelsize': 16,
+          'axes.titlesize': 14,
+          'font.size': 8,
+          'legend.fontsize': 14,
+          'xtick.labelsize': 16,
+          'ytick.labelsize': 16,
+          'figure.figsize': (10, 10),}
+#          'text.usetex': True}
+
+from matplotlib import rc
+# plt.rc('text.latex', preamble=r'\usepackage{braket}')
+plt.rcParams.update(params)
+
+N_arr = np.arange(6, 14, 2)
+
+path = os.getcwd()
+
+data_path = path + '/exp_data/'
+
+def compare_two_expect_all(task, prec, pkg_str):
+    fig, ax = plt.subplots()
+
+    comp_mat = np.zeros((len(pkg_str), len(pkg_str)))
+
+    for p1 in range(len(pkg_str)):
+
+        dat_file_p1 = data_path + '{}/{}/exp_mat_{}_{}.npy'.format(task, pkg_str[p1], pkg_str[p1], prec)
+        # print(dat_file_p1)
+
+        for p2 in range(len(pkg_str)):
+
+            dat_file_p2 = data_path + '{}/{}/exp_mat_{}_{}.npy'.format(task, pkg_str[p2], pkg_str[p2], prec)
+
+            if os.path.isfile(dat_file_p1) and os.path.isfile(dat_file_p2):
+
+                dat_p1 = np.load(dat_file_p1)
+                dat_p2 = np.load(dat_file_p2)
+
+                comp_mat[p1, p2] = np.log10(np.sum(np.abs(dat_p1 - dat_p2))/float(16))
+                # print(comp_mat[p1, p2])
+            else:
+                continue
+
+    plt.imshow(comp_mat, vmin=-16, vmax=0)
+
+    plt.xticks(range(len(pkg_str)), pkg_str)
+    locs, labels = plt.xticks()
+
+    plt.setp(labels, rotation=90)
+    plt.yticks(range(len(pkg_str)), pkg_str)
+    locs, labels = plt.yticks()
+    plt.colorbar()
+    plt.tight_layout()
+    # plt.savefig(fn)
+    plt.show()
+
+package_str_sp = ['cirq', 'hybridq', 'intel_qs_cpp', 'pennylane', 'pennylane_l', 'qibo', 'qibojit', 'qiskit', 'qsimcirq', 'quest', 'yao', 'qiskit_gpu', 'hybridq_gpu', 'qibojit_gpu', 'qsimcirq_gpu', 'yao_gpu']#, 'pennylane_l_gpu', 'quest_gpu', 'projectq', 'qulacs', 'qulacs_gpu']
+
+sorted_package_str_sp = ['qiskit_gpu', 'cuquantum_qiskit', 'qiskit', 'pennylane', 'cirq', 'qsimcirq', 'qsimcirq_gpu', 'qibojit_gpu', 'qibojit', 'qibo', 'intel_qs_cpp', 'quest', 'cuquantum_qsimcirq', 'pennylane_l', 'hybridq_gpu', 'hybridq', 'yao', 'yao_gpu']
+
+# compare_two_expect_all('hdyn', 'sp', sorted_package_str_sp)#, 'compare_packages_qft_sp_N16_log.pdf')
+
+package_str_dp = ['cirq', 'intel_qs_cpp', 'pennylane', 'pennylane_l', 'qibo', 'qibojit', 'qiskit', 'quest', 'yao', 'qiskit_gpu', 'qibojit_gpu', 'yao_gpu', 'pennylane_l_gpu', 'quest_gpu', 'projectq', 'qulacs', 'qulacs_gpu']#, 'hybridq', 'qsimcirq', 'hybridq_gpu', 'qsimcirq_gpu']
+
+sorted_package_str_dp = sorted_package_str_dp = ['qibojit_gpu', 'qulacs', 'qiskit_gpu', 'cuquantum_qiskit', 'qiskit', 'qpanda', 'intel_qs_cpp', 'quest', 'qibo', 'qibojit', 'pennylane_l_gpu', 'pennylane', 'yao_gpu', 'yao', 'pennylane_l', 'cirq', 'quest_gpu', 'projectq', 'qulacs_gpu']
+
+# compare_two_expect_all('hdyn', 'dp', sorted_package_str_dp)#, 'compare_packages_qft_dp_N16_log.pdf')

plot_scripts/plot_display_1v1.py

@@ -2,8 +2,10 @@ import numpy as np
 import h5py
 import os
 
+import mercury as mr
+
 import sys
-sys.path.append('/plot_scripts')
+sys.path.append('/plot_scripts/')
 from map_packages_colors_1v1 import *
 from plot_scripts_1v1 import *
 
@@ -86,6 +88,8 @@ def abs_time(t1, p1, p1_cc, p1_pr, t2, p2, p2_cc, p2_pr, N_end):
 
     fig, ax = plt.subplots()
 
+    mr.Md(f"TtS performance of the selected options")
+
     if os.path.isfile(data_file_p1) and os.path.isfile(data_file_p2):
             h5f_1 = h5py.File(data_file_p1, 'r')
             dat_1 = h5f_1[storage_dict[p1]][:]
@@ -108,14 +112,14 @@ def abs_time(t1, p1, p1_cc, p1_pr, t2, p2, p2_cc, p2_pr, N_end):
 
             gen_settings(fig, ax, r"N (system size)", r"Time ($t_{package}$)", False, True, True, N_arr[0]-2, N_arr[-1], True, 10**-1, 10**5, "out", None)
     else:
-        print(" Re-select the options as the requested option data is not available.")
+        mr.Md(f" Re-select the options as the requested configuration is not supported (check the table in the index page for supported configurations)")
 
 # abs_time(task_1, package_1, p1_com_cap, p1_prec, task_2, package_2, p2_com_cap, p2_prec)
 
 def relative_time_wrt_pack(t1, p1, p1_cc, p1_pr, t2, p2, p2_cc, p2_pr, N_end):
 
-    print("-----------------------------------------------------")
-    print("Relative time")
+    mr.Md("___")
+    mr.Md(f"Relative performance")
 
     if t1 == "Heisenberg dynamics":
         t1 = "hdyn"

plot_scripts/plot_display_all.py

@@ -2,8 +2,10 @@ import numpy as np
 import h5py
 import os
 
+import mercury as mr
+
 import sys
-sys.path.append('/plot_scripts')
+sys.path.append('/plot_scripts/')
 from map_packages_colors_all import *
 from plot_scripts_all import *
 
@@ -51,6 +53,8 @@ def abs_time(t, cc, pr, compare_pack, N_end):
 
     pack_list = []
 
+    mr.Md(f"TtS performance of the different packages")
+
     for package in package_str:
         data_file = dir + '/data/{}/{}_{}_{}.h5'.format(t, package, cc, pr)
 
@@ -75,11 +79,12 @@ def abs_time(t, cc, pr, compare_pack, N_end):
     #     print(" Re-select the options as the requested option data is not available.")
 
     if compare_pack not in pack_list:
-        print("Select one from the list: ", pack_list)
+        mr.Md(f"For relative performance select one from the list: , {pack_list}")
+        # mr.Md(" Select one package from the list ")
 
     else:
-        print(" -------------------------------------------------------- ")
-        print(" Comparison to the selected package ")
+        mr.Md("___")
+        mr.Md(f"Relative performance to the package {compare_pack}")
 
         fig, ax = plt.subplots()
 

plot_scripts/plot_display_com_pack.py

@@ -2,8 +2,10 @@ import numpy as np
 import h5py
 import os
 
+import mercury as mr
+
 import sys
-sys.path.append('/plot_scripts')
+sys.path.append('/plot_scripts/')
 from map_packages_colors_all import *
 from plot_scripts_all import *
 
@@ -117,8 +119,8 @@ def abs_time_pack(t1, t2, cc, pr, N_end):
     else:
         N_arr = N_arr_t1
 
-    print(" Number of gates of Task 1: ", ng1)
-    print(" Number of gates of Task 2: ", ng2)
+    mr.Md(f" Number of gates of Task 1: {ng1}")
+    mr.Md(f" Number of gates of Task 2: {ng2}")
 
     plot_comp_data_n_arr(N_arr, ng1, ng2, 'gate_count')
     gen_settings(fig, ax, r"N (system size)", r"Task I/Task II ($t_{T1}/t_{T2}$)", False, False, True, N_arr[0]-2, N_arr[-1], True, -3, 25, "out", None)

plot_scripts/plot_display_mgpu.py

@@ -2,8 +2,10 @@ import numpy as np
 import h5py
 import os
 
+import mercury as mr
+
 import sys
-sys.path.append('/plot_scripts')
+sys.path.append('/plot_scripts/')
 from map_packages_colors_mgpu import *
 from plot_scripts_mgpu import *
 
@@ -29,6 +31,8 @@ def abs_time(t, pr, n_gpu, compare_pack, N_end):
 
     pack_list = []
 
+    mr.Md(f"TtS performance of the different packages")
+
     for package in package_str:
         data_file = dir + '/data/{}/{}_gpu_{}_{}.h5'.format(t, package, n_gpu, pr)
 
@@ -56,6 +60,9 @@ def abs_time(t, pr, n_gpu, compare_pack, N_end):
     main_dat = h5f[storage_dict[compare_pack]][:]
     h5f.close()
 
+    mr.Md("___")
+    mr.Md(f"Relative performance to the package {compare_pack}")
+
     for package in package_str:
         dat_file = dir + '/data/{}/{}_gpu_{}_{}.h5'.format(t, package, n_gpu, pr)
         if os.path.isfile(dat_file):
@@ -91,6 +98,8 @@ def abs_time_ngpus(t, pr, pack, compare_ngpu, N_end):
     else:
         N_arr = range(12, N_end, 2)
 
+    mr.Md(f"TtS performance of the package with respect to different number of GPU's")
+
     for ngpu in [1, 2, 4, 8]:
 
         data_file = dir + '/data/{}/{}_gpu_{}_{}.h5'.format(t, pack, ngpu, pr)
@@ -108,6 +117,8 @@ def abs_time_ngpus(t, pr, pack, compare_ngpu, N_end):
     # else:
     #     print(" Re-select the options as the requested option data is not available.")
 
+    mr.Md("___")
+    mr.Md(f"Relative performance to the {compare_ngpu} GPUs")
 
     fig, ax = plt.subplots()