Skip to content

GitLab

Unverified Commit d947dee0 authored by moto-meta's avatar moto-meta Committed by GitHub
Browse files

Resolve lint issues 

Differential Revision: D50205775

Pull Request resolved: https://github.com/pytorch/audio/pull/3651
parent 57f7f522
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
src/libtorio/ffmpeg/stream_reader/conversion.cpp
View file @ d947dee0
...@@ -98,6 +98,7 @@ torch::Tensor get_image_buffer( ...@@ -98,6 +98,7 @@ torch::Tensor get_image_buffer(
shape, torch::TensorOptions().dtype(dtype).layout(torch::kStrided)); shape, torch::TensorOptions().dtype(dtype).layout(torch::kStrided));
} }
#ifdef USE_CUDA
torch::Tensor get_image_buffer( torch::Tensor get_image_buffer(
at::IntArrayRef shape, at::IntArrayRef shape,
torch::Device device, torch::Device device,
...@@ -109,6 +110,7 @@ torch::Tensor get_image_buffer( ...@@ -109,6 +110,7 @@ torch::Tensor get_image_buffer(
.layout(torch::kStrided) .layout(torch::kStrided)
.device(device)); .device(device));
} }
#endif // USE_CUDA
} // namespace } // namespace
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment