Update if nvjpeg not found

pytorch · Jun 5, 2024 · b3d06cb · b3d06cb
1 parent 62e072a
commit b3d06cb
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/benchmarks/encoding.py b/benchmarks/encoding.py
@@ -1,9 +1,6 @@
 import os
 import platform
 import statistics
-import tarfile
-import tempfile
-import urllib.request
 
 import torch
 import torch.utils.benchmark as benchmark
@@ -52,7 +49,7 @@ def run_benchmark(batch):
                         stmt=stmt,
                         setup="import torchvision",
                         globals={"batch_input": batch_input},
-                        label=f"Image Encoding",
+                        label="Image Encoding",
                         sub_label=f"{device.upper()} ({strat}): {stmt}",
                         description=f"{size} images",
                         num_threads=num_threads,

diff --git a/torchvision/csrc/io/image/cuda/encode_jpegs_cuda.cpp b/torchvision/csrc/io/image/cuda/encode_jpegs_cuda.cpp
@@ -1,11 +1,15 @@
 #include "encode_jpegs_cuda.h"
 #if !NVJPEG_FOUND
+namespace vision {
+namespace image {
 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");
 }
+} // namespace image
+} // namespace vision
 #else
 
 #include <ATen/ATen.h>