Sync

Signed-off-by: Albert Wolant <[email protected]>
NVIDIA · Jun 10, 2024 · 2264892 · 2264892
1 parent dffd5b7
commit 2264892
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/plugins/video/pkg_src/src/decoder/video_decoder_mixed.cc b/plugins/video/pkg_src/src/decoder/video_decoder_mixed.cc
@@ -101,6 +101,7 @@ void VideoDecoderMixed::Run(dali::Workspace &ws) {
 
       for (int i = 0; i < nFrameReturned; i++) {
         pFrame = sample.decoder_->GetFrame();
+        CUDA_CALL(cudaStreamSynchronize(cuStream));
 
         uint8_t *dpFrame = output_data + num_frames * sample.demuxer_->GetHeight() *
                                              sample.demuxer_->GetWidth() * 3;