From c7d41845c088a331e891015e4490f33fa811663c Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Wed, 4 Dec 2019 15:49:02 +0100
Subject: [PATCH] Also add TimeloopRuns
---
pystencils/datahandling/graph_datahandling.py | 1 +
pystencils_tests/test_graph_datahandling.py | 3 +++
2 files changed, 4 insertions(+)
diff --git a/pystencils/datahandling/graph_datahandling.py b/pystencils/datahandling/graph_datahandling.py
index 4c9b635b7..5d02946a3 100644
--- a/pystencils/datahandling/graph_datahandling.py
+++ b/pystencils/datahandling/graph_datahandling.py
@@ -110,6 +110,7 @@ class GraphDataHandling(pystencils.datahandling.SerialDataHandling):
argument_list = [argument_list]
def run(self, time_steps=1):
+ self.parent.call_queue.append(TimeloopRun(self, time_steps))
super().run(time_steps)
def __init__(self, *args, **kwargs):
diff --git a/pystencils_tests/test_graph_datahandling.py b/pystencils_tests/test_graph_datahandling.py
index ccb0ebf91..aefa51a45 100644
--- a/pystencils_tests/test_graph_datahandling.py
+++ b/pystencils_tests/test_graph_datahandling.py
@@ -71,3 +71,6 @@ def test_graph_datahandling():
opt_params = {'target': 'gpu', 'gpu_indexing_params': {'block_size': (8, 4, 2)}}
lbm_step: LatticeBoltzmannStep = ldc_setup(domain_size=(10, 15), optimization=opt_params)
print(lbm_step._data_handling)
+
+ # Hier stehen alle aufgerufenen Aktionen drinnen. Die Klassen kommen aus pystencils.datahandling.graph_datahandling
+ print(lbm_step._data_handling.call_queue)
--
GitLab