diff --git a/pySDC/implementations/datatype_classes/cupy_mesh.py b/pySDC/implementations/datatype_classes/cupy_mesh.py index 01b9dde1e8..0b27524f32 100644 --- a/pySDC/implementations/datatype_classes/cupy_mesh.py +++ b/pySDC/implementations/datatype_classes/cupy_mesh.py @@ -1,5 +1,4 @@ import cupy as cp -from pySDC.core.errors import DataError try: from mpi4py import MPI @@ -37,6 +36,7 @@ def __new__(cls, init, val=0.0, **kwargs): ): obj = cp.ndarray.__new__(cls, init[0], dtype=init[2], **kwargs) obj.fill(val) + cls.comm = init[1] else: raise NotImplementedError(type(init)) return obj