Fix confusing name

Author: NicolasHug

src/torchcodec/_core/CustomNvdecDeviceInterface.cpp

@@ -381,25 +381,25 @@ void CustomNvdecDeviceInterface::convertAVFrameToFrameOutput(
     std::optional<torch::Tensor> preAllocatedOutputTensor) {
   // For custom NVDEC, the frame should already be on GPU
   // We need to convert from NVDEC's output format (typically NV12) to RGB
-  printf("OKAY\n");
+  printf("In CNI convertAVFrameToFrameOutput\n");
   fflush(stdout);
 
   TORCH_CHECK(
       avFrame->format == AV_PIX_FMT_CUDA,
       "Expected CUDA format frame from custom NVDEC decoder");
 
-  auto cpuDevice = torch::Device(torch::kCUDA);
-  auto cpuInterface = createDeviceInterface(cpuDevice);
+  auto cudaDevice = torch::Device(torch::kCUDA);
+  auto cudaInterface = createDeviceInterface(cudaDevice);
 
-  FrameOutput cpuFrameOutput;
-  cpuInterface->convertAVFrameToFrameOutput(
+  FrameOutput cudaFrameOutput;
+  cudaInterface->convertAVFrameToFrameOutput(
       videoStreamOptions,
       timeBase,
       avFrame,
-      cpuFrameOutput,
+      cudaFrameOutput,
       preAllocatedOutputTensor);
 
-  frameOutput.data = cpuFrameOutput.data.to(device_);
+  frameOutput.data = cudaFrameOutput.data.to(device_);
 }
 
 } // namespace facebook::torchcodec

src/torchcodec/_core/SingleStreamDecoder.cpp

@@ -1333,8 +1333,6 @@ FrameOutput SingleStreamDecoder::convertAVFrameToFrameOutput(
   if (streamInfo.avMediaType == AVMEDIA_TYPE_AUDIO) {
     convertAudioAVFrameToFrameOutputOnCPU(avFrame, frameOutput);
   } else if (deviceInterface_) {
-    printf("Calling device interface to convert AVFrame to FrameOutput\n");
-    fflush(stdout);
     deviceInterface_->convertAVFrameToFrameOutput(
         streamInfo.videoStreamOptions,
         streamInfo.timeBase,