Spaces:
Runtime error
Runtime error
| import numpy as np | |
| import h5py | |
| import os | |
| import mercury as mr | |
| import sys | |
| sys.path.append('/plot_scripts/') | |
| from map_packages_colors_1v1 import * | |
| from plot_scripts_1v1 import * | |
| # package_str = ['qiskit' , 'cirq', 'qsimcirq', 'pennylane', 'pennylane_l', 'qibo', 'qibojit', 'yao', 'quest', 'qulacs', 'intel_qs_cpp', 'projectq', 'svsim', 'hybridq', 'hiq', 'qcgpu', 'qrack_sch', 'cuquantum_qiskit', 'cuquantum_qsimcirq', 'qpanda'] | |
| def abs_time_pack(task, package, pr, N_end): | |
| if task == "Heisenberg dynamics": | |
| task = "hdyn" | |
| elif task == "Random Quantum Circuit": | |
| task = "rqc" | |
| elif task == "Quantum Fourier Transform": | |
| task = "qft" | |
| if pr == "Single": | |
| pr = "sp" | |
| elif pr == "Double": | |
| pr = "dp" | |
| fig, ax = plt.subplots() | |
| dir = os.getcwd() | |
| if task == 'hdyn' or task == 'qft': | |
| N_arr = np.arange(6, N_end, 2) | |
| elif task == 'rqc': | |
| N_arr = np.arange(12, N_end, 2) | |
| dat_fst = dir + '/data/{}/{}_singlethread_{}.h5'.format(task, package, pr) | |
| dat_fmt = dir + '/data/{}/{}_multithread_{}.h5'.format(task, package, pr) | |
| dat_fgpu = dir + '/data/{}/{}_gpu_{}.h5'.format(task, package, pr) | |
| if not os.path.isfile(dat_fst) and not os.path.isfile(dat_fmt) and not os.path.isfile(dat_fgpu): | |
| return mr.Md(f"Precision {pr} possibly not supported") | |
| mr.Md(f"TtS performance of simulation packages with different compute capabilities") | |
| if os.path.isfile(dat_fst): | |
| h5f_st = h5py.File(dat_fst, 'r') | |
| dat_st = h5f_st[storage_dict[package]][:] | |
| h5f_st.close() | |
| plot_abs_data_n_arr(N_arr, dat_st, package+'_'+task+'_singlethread_'+pr) | |
| if os.path.isfile(dat_fmt): | |
| h5f_mt = h5py.File(dat_fmt, 'r') | |
| dat_mt = h5f_mt[storage_dict[package]][:] | |
| h5f_mt.close() | |
| plot_abs_data_n_arr(N_arr, dat_mt, package+'_'+task+'_multithread_'+pr) | |
| if os.path.isfile(dat_fgpu): | |
| h5f_gpu = h5py.File(dat_fgpu, 'r') | |
| dat_gpu = h5f_gpu[storage_dict[package]][:] | |
| h5f_gpu.close() | |
| plot_abs_data_n_arr(N_arr, dat_gpu, package+'_'+task+'_gpu_'+pr) | |
| 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) | |
| mr.Md("___") | |
| mr.Md(f"Relative performance to singlethread performance") | |
| fig, ax = plt.subplots() | |
| if os.path.isfile(dat_fst) and os.path.isfile(dat_fmt): | |
| plot_comp_data_n_arr(N_arr, dat_st, dat_st, package+'_'+task+'_singlethread_'+pr) | |
| plot_comp_data_n_arr(N_arr, dat_st, dat_mt, package+'_'+task+'_multithread_'+pr) | |
| if os.path.isfile(dat_fst) and os.path.isfile(dat_fgpu): | |
| plot_comp_data_n_arr(N_arr, dat_st, dat_gpu, package+'_'+task+'_gpu_'+pr) | |
| gen_settings(fig, ax, r"N (system size)", r"Relative to singlethread", False, True, True, N_arr[0]-2, N_arr[-1], True, 10**-1, 10**3, "out", None) | |
| mr.Md("___") | |
| mr.Md(f"Relative performance to multithread performance") | |
| fig, ax = plt.subplots() | |
| if os.path.isfile(dat_fmt) and os.path.isfile(dat_fgpu): | |
| plot_comp_data_n_arr(N_arr, dat_mt, dat_gpu, package+'_'+task+'_gpu_'+pr) | |
| plot_comp_data_n_arr(N_arr, dat_mt, dat_mt, package+'_'+task+'_multithread_'+pr) | |
| gen_settings(fig, ax, r"N (system size)", r"Relative to multithread", False, True, True, N_arr[0]-2, N_arr[-1], True, 10**-1, 10**2, "out", None) | |
| # else: | |
| # print(" Re-select the options as the requested option data is not available.") | |
| # pkg_str = ['qiskit' , 'cirq', 'qsimcirq', 'pennylane', 'pennylane_l', 'qibo', 'qibojit', 'yao', 'quest', 'qulacs', 'intel_qs_cpp', 'projectq', 'svsim', 'hybridq', 'hiq', 'qcgpu', 'qrack_sch'] | |
| # abs_time_pack("Heisenberg dynamics", 'qsimcirq', 'Double', 36) | |
| # abs_time(pkg_str, task_1, p_com_cap, p_prec) | |
| # abs_time("Heisenberg dynamics", "Singlethread", "Single", 'qsimcirq') | |
| # abs_time_pack("Heisenberg dynamics", "Random Quantum Circuit", "Singlethread", "Single", 34) | |
| # abs_time_pack("Heisenberg dynamics", "Quantum Fourier Transform", "GPU", "Single", 38) | |