only synchronize "main" stream

Author: jorenretel

bigwig_loader/streamed_dataset.py

@@ -251,8 +251,7 @@ def _generate_batches(self) -> Generator[Batch, None, None]:
 
                 for select in self._slices_objects(n_samples, slice_size):
                     with self.main_stream as stream:
-                        # stream.synchronize()
-                        cp.cuda.Device().synchronize()
+                        stream.synchronize()
 
                         values = intervals_to_values(
                             array_start=start_data,