[FFmpeg-devel] [PATCH FFmpeg 14/15] configure: libtorch cuda check with new HAVE variable HAVE_LIBTORCH_CUDA

[FFmpeg-devel] [PATCH FFmpeg 14/15] configure: libtorch cuda check with new HAVE variable HAVE_LIBTORCH_CUDA

[m.kaindl0208]

Try the new filters using my Github Repo https://github.com/MaximilianKaindl/DeepFFMPEGVideoClassification. 

Any Feedback is appreciated!

Signed-off-by: MaximilianKaindl <m.kaindl0208@gmail.com>
---
 configure | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/configure b/configure
index 2c08901adc..b2a1948855 100755
--- a/configure
+++ b/configure
@@ -2527,6 +2527,7 @@ HAVE_LIST="
     $TYPES_LIST
     gzip
     libdrm_getfb2
+    libtorch_cuda
     makeinfo
     makeinfo_html
     opencl_d3d11
@@ -7042,6 +7043,7 @@ enabled libtensorflow     && require libtensorflow tensorflow/c/c_api.h TF_Versi
 enabled libtesseract      && require_pkg_config libtesseract tesseract tesseract/capi.h TessBaseAPICreate
 enabled libtheora         && require libtheora theora/theoraenc.h th_info_init -ltheoraenc -ltheoradec -logg
 enabled libtls            && require_pkg_config libtls libtls tls.h tls_configure
+enabled libtorch          && check_cxxflags -std=c++17 && require_cpp libtorch torch/torch.h "torch::Tensor" -ltorch -lc10 -ltorch_cpu -lstdc++ -lpthread && { check_lib_cpp libtorch torch/cuda.h "" "-Wl,--no-as-needed -lc10 -ltorch -ltorch_cpu -lc10_cuda -ltorch_cuda -lstdc++ -lpthread" && enable libtorch_cuda; }
 enabled libtorch          && check_cxxflags -std=c++17 && require_cpp libtorch torch/torch.h "torch::Tensor" -ltorch -lc10 -ltorch_cpu -lstdc++ -lpthread
 enabled libtwolame        && require libtwolame twolame.h twolame_init -ltwolame &&
                              { check_lib libtwolame twolame.h twolame_encode_buffer_float32_interleaved -ltwolame ||
-- 
2.34.1


_______________________________________________
ffmpeg-devel mailing list
ffmpeg-devel@ffmpeg.org
https://ffmpeg.org/mailman/listinfo/ffmpeg-devel

To unsubscribe, visit link above, or email
ffmpeg-devel-request@ffmpeg.org with subject "unsubscribe".