Skip to content

Commit fd2d3d5

Browse files
authored
[Cherry-pick for 0.1.1] Improve GPU frames accuracy (#413) (#419)
1 parent cdd768d commit fd2d3d5

File tree

5 files changed

+92
-137
lines changed

5 files changed

+92
-137
lines changed

src/torchcodec/decoders/_core/CudaDevice.cpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -225,7 +225,7 @@ void convertAVFrameToDecodedOutputOnCuda(
225225
auto start = std::chrono::high_resolution_clock::now();
226226
NppStatus status;
227227
if (src->colorspace == AVColorSpace::AVCOL_SPC_BT709) {
228-
status = nppiNV12ToRGB_709HDTV_8u_P2C3R(
228+
status = nppiNV12ToRGB_709CSC_8u_P2C3R(
229229
input,
230230
src->linesize[0],
231231
static_cast<Npp8u*>(dst.data_ptr()),

0 commit comments

Comments
 (0)