diff --git a/pystencils/datahandling/graph_datahandling.py b/pystencils/datahandling/graph_datahandling.py index 4c9b635b75998ddcab299bfbb005ac3e0ef5af8f..5d02946a3d7d7032c0e9a99cfb69ccee214359c3 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 ccb0ebf914580cbfadcf5ea3c900ca956fb06b44..aefa51a45ae1e39a75d0e65839d2d4da680bbbd1 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)