Let simulator return executed cycles

wallento · Jun 25, 2020 · 534fc87 · 534fc87
1 parent 4e993dc
commit 534fc87
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/riscvmodel/sim.py b/riscvmodel/sim.py
@@ -17,11 +17,14 @@ def load_data(self, data = "", *, address=0):
 
   def run(self, *, pc=0):
     self.model.reset(pc=pc)
+    cnt = 0
     while True:
       try:
         self.model.issue(self.program[int(self.model.state.pc)>>2])
+        cnt += 1
       except IndexError as e:
-        return
+        return cnt
+    return cnt
 
   def dump_data(self, *, address=0, size=None):
     data = b""