From 9ee11a3c2f11462a5743c3e45030bb8397e6822e Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Tue, 10 Dec 2019 13:10:05 +0100
Subject: [PATCH] Add TimeloopRun

---
 pystencils/datahandling/graph_datahandling.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/pystencils/datahandling/graph_datahandling.py b/pystencils/datahandling/graph_datahandling.py
index 5d02946a..adae1021 100644
--- a/pystencils/datahandling/graph_datahandling.py
+++ b/pystencils/datahandling/graph_datahandling.py
@@ -80,6 +80,10 @@ class TimeloopRun:
                 + '\nPost:\n'
                 + '\n   '.join(str(f) for f in self.timeloop._post_run_functions))
 
+    @property
+    def asts(self):
+        return self.timeloop._single_step_asts
+
 
 class GraphDataHandling(pystencils.datahandling.SerialDataHandling):
 
-- 
GitLab