Update if nvjpeg not found

pytorch · Jun 5, 2024 · 7aef769 · 7aef769
1 parent 62e072a
commit 7aef769
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchvision/csrc/io/image/cuda/encode_jpegs_cuda.cpp b/torchvision/csrc/io/image/cuda/encode_jpegs_cuda.cpp
@@ -1,7 +1,7 @@
 #include "encode_jpegs_cuda.h"
 #if !NVJPEG_FOUND
 std::vector<torch::Tensor> encode_jpegs_cuda(
-    const std::vector<torch::Tensor>& images,
+    const std::vector<torch::Tensor>& decoded_images,
     const int64_t quality) {
   TORCH_CHECK(
       false, "encode_jpegs_cuda: torchvision not compiled with nvJPEG support");