We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 333d0ef commit a83f4faCopy full SHA for a83f4fa
torchvision/csrc/io/image/cuda/decode_jpegs_cuda.cpp
@@ -13,7 +13,6 @@ std::vector<torch::Tensor> decode_jpegs_cuda(
13
} // namespace vision
14
15
#else
16
-#include <ATen/ATen.h>
17
#include <ATen/cuda/CUDAContext.h>
18
#include <ATen/cuda/CUDAEvent.h>
19
#include <c10/cuda/CUDAGuard.h>
torchvision/csrc/io/image/cuda/encode_jpegs_cuda.cpp
@@ -12,11 +12,8 @@ std::vector<torch::Tensor> encode_jpegs_cuda(
12
-#include <cuda_runtime.h>
-#include <torch/nn/functional.h>
20
#include <iostream>
21
#include <memory>
22
#include <string>
0 commit comments