Set the output data type for image reconstruction to float32.

api/python/arrus/utils/imaging.py

@@ -663,7 +663,7 @@ def gpu_convolve1d(data):
                 shared_memory_size)
             return fir_output_buffer
         self.convolve1d_func = gpu_convolve1d
-        return const_metadata
+        return const_metadata.copy(dtype=self.xp.float32)
 
     def process(self, data):
         return self.convolve1d_func(data)
@@ -2038,6 +2038,7 @@ def prepare(self, const_metadata: arrus.metadata.ConstMetadata):
         n_samples = next(iter(n_samples_set))
         batch_size = fcm.batch_size
         self.output_shape = (batch_size, n_frames, n_samples, n_channels)
+        print(self.output_shape)
         self._output_buffer = xp.zeros(shape=self.output_shape, dtype=xp.int16)
         if xp == np:
             # CPU