From 47100c5c4e018ad9a9942ae62e59dba9b101358c Mon Sep 17 00:00:00 2001
From: Florent Kermarrec <florent@enjoy-digital.fr>
Date: Mon, 23 Nov 2020 17:36:54 +0100
Subject: [PATCH] bench/test_etherbone: add initial speed_test.

---
 bench/test_etherbone.py | 33 +++++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)

diff --git a/bench/test_etherbone.py b/bench/test_etherbone.py
index e69bbe8..8046616 100755
--- a/bench/test_etherbone.py
+++ b/bench/test_etherbone.py
@@ -14,6 +14,9 @@ import argparse
 
 from litex import RemoteClient
 
+KiB = 1024
+MiB = 1024*KiB
+
 # Identifier Test ----------------------------------------------------------------------------------
 
 def ident_test(port):
@@ -95,6 +98,32 @@ def sram_test(port):
 
     wb.close()
 
+# Speed Test ---------------------------------------------------------------------------------------
+
+def speed_test(port):
+    wb = RemoteClient(port=port)
+    wb.open()
+
+    test_size = 16*KiB
+
+    print("Testing write speed... ", end="")
+    start = time.time()
+    for i in range(test_size//4):
+        wb.write(wb.mems.sram.base + i%0x100, i)
+    end = time.time()
+    duration = (end - start)
+    print("{:3.2f} KiB/s".format(test_size/(duration*KiB)))
+
+    print("Testing read speed... ", end="")
+    start = time.time()
+    for i in range(test_size//4):
+        wb.read(wb.mems.sram.base + i%0x100)
+    end = time.time()
+    duration = (end - start)
+    print("{:3.2f} KiB/s".format(test_size/(duration*KiB)))
+
+    wb.close()
+
 # Run ----------------------------------------------------------------------------------------------
 
 def main():
@@ -103,6 +132,7 @@ def main():
     parser.add_argument("--ident",  action="store_true", help="Read FPGA identifier")
     parser.add_argument("--access", action="store_true", help="Test single Write/Read access over Etherbone")
     parser.add_argument("--sram",   action="store_true", help="Test SRAM access over Etherbone")
+    parser.add_argument("--speed",  action="store_true", help="Test speed over Etherbone")
     args = parser.parse_args()
 
     port = int(args.port, 0)
@@ -116,5 +146,8 @@ def main():
     if args.sram:
         sram_test(port=port)
 
+    if args.speed:
+        speed_test(port=port)
+
 if __name__ == "__main__":
     main()