Skip to content
Snippets Groups Projects
Commit f6907740 authored by Stephan Seitz's avatar Stephan Seitz
Browse files

Add DataTransfer.__str__, Swap.__str__

parent 9ee11a3c
No related branches found
No related tags found
1 merge request!98WIP: Graph datahandling
...@@ -42,6 +42,9 @@ class DataTransfer: ...@@ -42,6 +42,9 @@ class DataTransfer:
self.field = field self.field = field
self.kind = kind self.kind = kind
def __str__(self):
return f'DataTransferKind: {self.kind} with {self.field}'
class Swap(DataTransfer): class Swap(DataTransfer):
def __init__(self, source, destination, gpu): def __init__(self, source, destination, gpu):
...@@ -49,6 +52,9 @@ class Swap(DataTransfer): ...@@ -49,6 +52,9 @@ class Swap(DataTransfer):
self.field = source self.field = source
self.destination = destination self.destination = destination
def __str__(self):
return f'Swap: {self.field} with {self.destination}'
class Communication(DataTransfer): class Communication(DataTransfer):
def __init__(self, field, stencil, gpu): def __init__(self, field, stencil, gpu):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment