diff --git a/pystencils/datahandling/graph_datahandling.py b/pystencils/datahandling/graph_datahandling.py index adae1021322e6a5df5f9885675a713b1d82aaac5..650bf51f6acf9f88c6660ef3ee388bd43a0daa47 100644 --- a/pystencils/datahandling/graph_datahandling.py +++ b/pystencils/datahandling/graph_datahandling.py @@ -42,6 +42,9 @@ class DataTransfer: self.field = field self.kind = kind + def __str__(self): + return f'DataTransferKind: {self.kind} with {self.field}' + class Swap(DataTransfer): def __init__(self, source, destination, gpu): @@ -49,6 +52,9 @@ class Swap(DataTransfer): self.field = source self.destination = destination + def __str__(self): + return f'Swap: {self.field} with {self.destination}' + class Communication(DataTransfer): def __init__(self, field, stencil, gpu):