Merge branch 'main' into lajenfljnaef

Author: NicolasHug

setup.py

@@ -127,7 +127,7 @@ def get_macros_and_flags():
             if NVCC_FLAGS is None:
                 nvcc_flags = []
             else:
-                nvcc_flags = nvcc_flags.split(" ")
+                nvcc_flags = NVCC_FLAGS.split(" ")
         extra_compile_args["nvcc"] = nvcc_flags
 
     if sys.platform == "win32":

torchvision/csrc/io/image/cpu/decode_webp.cpp

@@ -44,10 +44,12 @@ torch::Tensor decode_webp(
 
   auto decoded_data =
       decoding_func(encoded_data_p, encoded_data_size, &width, &height);
+
   TORCH_CHECK(decoded_data != nullptr, "WebPDecodeRGB[A] failed.");
 
+  auto deleter = [decoded_data](void*) { WebPFree(decoded_data); };
   auto out = torch::from_blob(
-      decoded_data, {height, width, num_channels}, torch::kUInt8);
+      decoded_data, {height, width, num_channels}, deleter, torch::kUInt8);
 
   return out.permute({2, 0, 1});
 }

torchvision/datasets/__init__.py

@@ -72,6 +72,7 @@
     "QMNIST",
     "MNIST",
     "KMNIST",
+    "MovingMNIST",
     "StanfordCars",
     "STL10",
     "SUN397",