Signed-off-by: MaximilianKaindl <m.kaindl0...@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
_______________________________________________
ffmpeg-devel mailing list
ffmpeg-devel@ffmpeg.org
https://ffmpeg.org/mailman/listinfo/ffmpeg-devel
To unsubscribe, visit link above, or email
ffmpeg-devel-requ...@ffmpeg.org with subject "unsubscribe".
