mirror of
https://git.ffmpeg.org/ffmpeg.git
synced 2026-04-18 08:30:24 +00:00
Add support for CUDA and ROCm (AMD GPU) devices in the LibTorch DNN backend. This works for both NVIDIA CUDA and AMD ROCm, as PyTorch exposes ROCm through the CUDA-compatible API. Usage: ./ffmpeg -i input.mp4 -vf scale=224:224,format=rgb24,dnn_processing=dnn_backend=torch:model=sr_model_torch.pt:device=cuda output.mp4 Reviewed-by: Guo Yejun <yejun.guo@intel.com> Signed-off-by: younengxiao <steven.xiao@amd.com> |
||
|---|---|---|
| .. | ||
| dnn_backend_common.c | ||
| dnn_backend_common.h | ||
| dnn_backend_openvino.c | ||
| dnn_backend_tf.c | ||
| dnn_backend_torch.cpp | ||
| dnn_interface.c | ||
| dnn_io_proc.c | ||
| dnn_io_proc.h | ||
| Makefile | ||
| queue.c | ||
| queue.h | ||
| safe_queue.c | ||
| safe_queue.h | ||