diff --git a/pySDC/implementations/datatype_classes/cupy_mesh.py b/pySDC/implementations/datatype_classes/cupy_mesh.py index 01b9dde1e8bc6d155bd58b04e23113065ba8efd3..0b27524f329245e8bdd07985b92bf82b30adaa2c 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 @@ class cupy_mesh(cp.ndarray): ): 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