From a4f181685e809f7f2ff12e95e8cf64e84c1b5b73 Mon Sep 17 00:00:00 2001
From: Maciej Dudek <mdudek@antmicro.com>
Date: Thu, 14 Nov 2024 14:38:05 +0100
Subject: [PATCH] Improve mem.py speed
Signed-off-by: Maciej Dudek <mdudek@antmicro.com>
---
rowhammer_tester/scripts/mem.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/rowhammer_tester/scripts/mem.py b/rowhammer_tester/scripts/mem.py
index 1bd541c9c..bb1155fb2 100755
--- a/rowhammer_tester/scripts/mem.py
+++ b/rowhammer_tester/scripts/mem.py
@@ -17,7 +17,7 @@ def memtest(wb, length, *, generator, base=None, verbose=None, burst=255):
if base is None:
base = wb.mems.main_ram.base
- refdata = [next(generator) for _ in range(length)]
+ refdata = list(itertools.islice(generator,length))
memwrite(wb, refdata, base=base, burst=burst)
data = memread(wb, length, base=base, burst=burst)