[FFmpeg-devel] [PATCH v2 FFmpeg 4/20] configure: libtorch CUDA support

[m.kaindl0208 at gmail.com]

Signed-off-by: MaximilianKaindl <m.kaindl0208 at gmail.com>
---
 configure | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/configure b/configure
index 7219faeaf2..ac074279cb 100755
--- a/configure
+++ b/configure
@@ -285,6 +285,7 @@ External library support:
   --enable-libtls          enable LibreSSL (via libtls), needed for https support
                            if openssl, gnutls or mbedtls is not used [no]
   --enable-libtorch        enable Torch as one DNN backend [no]
+  --enable-libtorch_cuda   enable Torch with CUDA Acceleration as one DNN backend [no]
   --enable-libtokenizers   enable tokenizers-cpp library [no]
   --enable-libtwolame      enable MP2 encoding via libtwolame [no]
   --enable-libuavs3d       enable AVS3 decoding via libuavs3d [no]
@@ -1968,6 +1969,7 @@ EXTERNAL_LIBRARY_LIST="
     libtesseract
     libtheora
     libtorch
+    libtorch_cuda
     libtokenizers
     libtwolame
     libuavs3d
@@ -2875,7 +2877,7 @@ dirac_parse_select="golomb"
 dovi_rpudec_select="golomb"
 dovi_rpuenc_select="golomb"
 dnn_deps="avformat swscale"
-dnn_deps_any="libtensorflow libopenvino libtorch libtokenizers"
+dnn_deps_any="libtensorflow libopenvino libtorch libtorch_cuda libtokenizers"
 error_resilience_select="me_cmp"
 evcparse_select="golomb"
 faandct_deps="faan"
@@ -7051,6 +7053,8 @@ 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
+enabled libtorch_cuda && enable libtorch && check_cxxflags -std=c++17 && require_cpp libtorch_cuda torch/torch.h "torch::Tensor" -ltorch -lc10 -ltorch_cpu -lstdc++ -lpthread && { require_cpp libtorch_cuda torch/cuda.h "" "-Wl,--no-as-needed -lc10 -ltorch -ltorch_cpu -lc10_cuda -ltorch_cuda -lstdc++ -lpthread"; }
 enabled libtokenizers     && check_cxxflags -std=c++17 && require_cpp libtokenizers "tokenizers_cpp.h tokenizers_c.h" "" -ltokenizers_cpp -ltokenizers_c -lstdc++
 enabled libtwolame        && require libtwolame twolame.h twolame_init -ltwolame &&
                              { check_lib libtwolame twolame.h twolame_encode_buffer_float32_interleaved -ltwolame ||
--
2.34.1