Skip to content

Commit 1a76abd

Browse files
committed
Small edits to benchmark
1 parent 88986ec commit 1a76abd

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

bench/ndarray/matmul_Blosc2PyTorch.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,6 @@ def build_dense_rowwarp_matrix(out_h=2000, in_h=2167,
8787
b2im[i:end] = res
8888
b2im[i + 1000, end + 1000] = res
8989
del dset
90-
print("Saved data to Blosc2.")
9190

9291
b2im = blosc2.open(urlpath="kevlar.b2nd", mode="r")
9392
b2im_trans = blosc2.open(urlpath="transform.b2nd", mode="r")
@@ -137,7 +136,7 @@ def build_dense_rowwarp_matrix(out_h=2000, in_h=2167,
137136
dset_b = f["data"]
138137
dset_out = f["out"]
139138

140-
for i in tqdm(range(0, len(dset_out), batch_size), desc="PyTorch Matmul"): # batch of 32
139+
for i in range(0, len(dset_out), batch_size): # batch of 32
141140
batch_a = torch.from_numpy(dset_a[i:i+batch_size]) # NumPy array slice
142141
batch_b = torch.from_numpy(dset_b[i:i+batch_size]) # NumPy array slice
143142
dset_out[i:i+batch_size] = torch.matmul(batch_a, batch_b)

0 commit comments

Comments
 (0)