From 9f69bc02c1cf319ec83d616b75746eb2c2dfda9c Mon Sep 17 00:00:00 2001
From: xiyehu2 <35537592+xiyehu2@users.noreply.github.com>
Date: Thu, 8 Dec 2022 11:43:51 -0600
Subject: [PATCH] keeping everything updated
---
Python/benchmarking.py | 10 ++++++----
1 file changed, 6 insertions(+), 4 deletions(-)
diff --git a/Python/benchmarking.py b/Python/benchmarking.py
index 1dc36e4..ec9e4db 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
)
--
GitLab