diff --git a/Python/benchmarking.py b/Python/benchmarking.py index 1dc36e4a3eb3ef0d50e0e9ebd8c2df44b10c5483..ec9e4db6ffbfaf7b23008649dd5d8f0e9e49ffc8 100644 --- a/Python/benchmarking.py +++ b/Python/benchmarking.py @@ -73,8 +73,6 @@ for i in range(nt+1): paths, off = get_rearrange_paths(f_idx, t_idx) nm[j] = count(paths, _table_cp, off) - # n_move[i,0] = np.mean(nm) - # n_move[i,1] = np.var(nm) n_move[i,0] = np.mean(nm) n_move[i,1] = np.var(nm) gpu_t[i,0] = np.mean(g_t) @@ -86,12 +84,16 @@ for i in range(nt+1): filling_ratio[i,0] = np.mean(ratio) filling_ratio[i,1] = np.var(ratio) +times = {} +times['gpu_t'] = gpu_t +times['cpu_t'] = cpu_t +times['total_t'] = total_t np.savez( - f"data/120722_benchmark_{nt}-half_wfm-only.npz", + f"data/120722_benchmark_{nt}-half_wfm-and-transfer.npz", wfm=twzr, target=target, filling_ratio=filling_ratio, - times=[gpu_t, cpu_t, total_t], + times=times, n_move=n_move )