Fixes

Author: brownbaerchen

pySDC/implementations/problem_classes/generic_spectral.py

@@ -133,10 +133,7 @@ def __init__(
             self.__heterogeneous_setup = False
 
     def heterogeneous_setup(self):
-        if self.heterogeneous and self.useGPU and not self.__heterogeneous_setup:
-            for key in ['BC_line_zero_matrix', 'BCs']:
-                setattr(self.spectral, key, getattr(self.spectral, key).get())
-
+        if self.heterogeneous and not self.__heterogeneous_setup:
             CPU_only = ['BC_line_zero_matrix', 'BCs']
             both = ['Pl', 'Pr', 'L', 'M']
 

pySDC/tests/test_problems/test_RayleighBenard3D.py

@@ -275,11 +275,11 @@ def test_banded_matrix(preconditioning):
 
 @pytest.mark.cupy
 def test_heterogeneous_implementation():
-    from pySDC.implementations.problem_classes.RayleighBenard3D import RayleighBenard3D, RayleighBenard3DHeterogeneous
+    from pySDC.implementations.problem_classes.RayleighBenard3D import RayleighBenard3D
 
     params = {'nx': 2, 'ny': 2, 'nz': 2, 'useGPU': True}
     gpu = RayleighBenard3D(**params)
-    het = RayleighBenard3DHeterogeneous(**params)
+    het = RayleighBenard3D(**params, heterogeneous=True)
 
     xp = gpu.xp