We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5bb3756 commit dd04495Copy full SHA for dd04495
src/torchcodec/_core/CudaDeviceInterface.cpp
@@ -414,17 +414,6 @@ std::optional<UniqueAVFrame> CudaDeviceInterface::convertTensorToAVFrame(
414
415
torch::Tensor hwcFrame = tensor.permute({1, 2, 0}).contiguous();
416
417
- at::cuda::CUDAStream currentStream =
418
- at::cuda::getCurrentCUDAStream(device_.index());
419
-
420
- nppCtx_->hStream = currentStream.stream();
421
- cudaError_t cudaErr =
422
- cudaStreamGetFlags(nppCtx_->hStream, &nppCtx_->nStreamFlags);
423
- TORCH_CHECK(
424
- cudaErr == cudaSuccess,
425
- "cudaStreamGetFlags failed: ",
426
- cudaGetErrorString(cudaErr));
427
428
NppiSize oSizeROI = {width, height};
429
NppStatus status = nppiRGBToNV12_8u_ColorTwist32f_C3P2R_Ctx(
430
static_cast<const Npp8u*>(hwcFrame.data_ptr()),
0 commit comments