Block to cpu function changes of base_block

httomo/base_block.py

@@ -111,7 +111,7 @@ def to_gpu(self):
         self._data = xp.asarray(self.data, order="C")
 
     def to_cpu(self):
-        if not gpu_enabled:
+        if not gpu_enabled or xp.get_array_module(self.data).__name__ == "numpy":
             return
         self._data = xp.asnumpy(self.data, order="C")
 

httomo/method_wrappers/generic.py

@@ -319,7 +319,10 @@ def execute(self, block: T) -> T:
             block = self._run_method(block, args)
             block = self._postprocess_data(block)
 
-        self._gpu_time_info.kernel = t.elapsed
+        if xp.get_array_module(block.data).__name__ == "cupy":
+            self._gpu_time_info.kernel = t.elapsed
+        else:
+            self._gpu_time_info.kernel = 0
 
         return block