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

[Guo, Yejun]

> -----Original Message-----
> From: ffmpeg-devel <ffmpeg-devel-bounces at ffmpeg.org> On Behalf Of
> m.kaindl0208 at gmail.com
> Sent: Tuesday, March 11, 2025 3:52 AM
> To: ffmpeg-devel at ffmpeg.org
> Subject: [FFmpeg-devel] [PATCH v2 FFmpeg 4/20] configure: libtorch CUDA
> support
> 
> 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

"enabled libtorch" is already in the file, why we need to add a new line here.


> torch/torch.h "torch::Tensor" -ltorch -lc10 -ltorch_cpu -lstdc++ -lpthread
> +enabled libtorch_cuda && enable libtorch && check_cxxflags -std=c++17

Please align the spaces between " enabled libtorch_cuda" and "&&" with others already in the file.

> +&& 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
> 
> 
> _______________________________________________
> ffmpeg-devel mailing list
> ffmpeg-devel at ffmpeg.org
> https://ffmpeg.org/mailman/listinfo/ffmpeg-devel
> 
> To unsubscribe, visit link above, or email ffmpeg-devel-request at ffmpeg.org
> with subject "unsubscribe".