Skip to content

Commit b32f44a

Browse files
committed
Try to build with CUDA support and FFmpeg4
1 parent 02dfeaf commit b32f44a

File tree

2 files changed

+2
-5
lines changed

2 files changed

+2
-5
lines changed

.github/workflows/linux_cuda_wheel.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,7 @@ jobs:
6868
python-version: ['3.9']
6969
cuda-version: ['11.8', '12.4', '12.6']
7070
# TODO: put back ffmpeg 5 https://github.com/pytorch/torchcodec/issues/325
71-
ffmpeg-version-for-tests: ['6', '7']
71+
ffmpeg-version-for-tests: ['4.4.2', '6', '7']
7272
container:
7373
image: "pytorch/manylinux2_28-builder:cuda${{ matrix.cuda-version }}"
7474
options: "--gpus all -e NVIDIA_DRIVER_CAPABILITIES=video,compute,utility"

src/torchcodec/decoders/_core/CMakeLists.txt

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -77,10 +77,7 @@ if(DEFINED ENV{BUILD_AGAINST_ALL_FFMPEG_FROM_S3})
7777
)
7878

7979

80-
if(NOT ENABLE_CUDA)
81-
# TODO: Enable more ffmpeg versions for cuda.
82-
make_torchcodec_library(libtorchcodec4 ffmpeg4)
83-
endif()
80+
make_torchcodec_library(libtorchcodec4 ffmpeg4)
8481
make_torchcodec_library(libtorchcodec7 ffmpeg7)
8582
make_torchcodec_library(libtorchcodec6 ffmpeg6)
8683
make_torchcodec_library(libtorchcodec5 ffmpeg5)

0 commit comments

Comments
 (0)