Resolve lint issues

Differential Revision: D50205775 Pull Request resolved: https://github.com/pytorch/audio/pull/3651

Resolve lint issues
Differential Revision: D50205775 Pull Request resolved: https://github.com/pytorch/audio/pull/3651
d947dee0 · moto-meta · GitHub · 57f7f522 · d947dee0
Unverified Commit d947dee0 authored Oct 12, 2023 by moto-meta Committed by GitHub Oct 12, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

src/libtorio/ffmpeg/stream_reader/conversion.cpp src/libtorio/ffmpeg/stream_reader/conversion.cpp +2 -0

No files found.
--- a/src/libtorio/ffmpeg/stream_reader/conversion.cpp
+++ b/src/libtorio/ffmpeg/stream_reader/conversion.cpp
@@ -98,6 +98,7 @@ torch::Tensor get_image_buffer(
      shape, torch::TensorOptions().dtype(dtype).layout(torch::kStrided));
 }

+#ifdef USE_CUDA
 torch::Tensor get_image_buffer(
    at::IntArrayRef shape,
    torch::Device device,
@@ -109,6 +110,7 @@ torch::Tensor get_image_buffer(
          .layout(torch::kStrided)
          .device(device));
 }
+#endif // USE_CUDA

 } // namespace