Spaces:
Runtime error
Runtime error
File size: 4,074 Bytes
d17e7e7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 |
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)
|