Skip to content

Commit dd04495

Browse files
committed
clean up unused npp stream setter
1 parent 5bb3756 commit dd04495

File tree

1 file changed

+0
-11
lines changed

1 file changed

+0
-11
lines changed

src/torchcodec/_core/CudaDeviceInterface.cpp

Lines changed: 0 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -414,17 +414,6 @@ std::optional<UniqueAVFrame> CudaDeviceInterface::convertTensorToAVFrame(
414414

415415
torch::Tensor hwcFrame = tensor.permute({1, 2, 0}).contiguous();
416416

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-
428417
NppiSize oSizeROI = {width, height};
429418
NppStatus status = nppiRGBToNV12_8u_ColorTwist32f_C3P2R_Ctx(
430419
static_cast<const Npp8u*>(hwcFrame.data_ptr()),

0 commit comments

Comments
 (0)