@@ -25,28 +25,29 @@ defaults:
25
25
26
26
jobs :
27
27
generate-matrix :
28
- uses : pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@main
28
+ uses : pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@release/2.8
29
29
with :
30
30
package-type : wheel
31
31
os : linux
32
32
test-infra-repository : pytorch/test-infra
33
- test-infra-ref : main
33
+ test-infra-ref : release/2.8
34
34
with-cpu : disable
35
35
with-xpu : disable
36
36
with-rocm : disable
37
37
with-cuda : enable
38
38
build-python-only : " disable"
39
+ python-versions : ' ["3.9", "3.10", "3.11", "3.12", "3.13"]'
39
40
build :
40
41
needs : generate-matrix
41
42
strategy :
42
43
fail-fast : false
43
44
name : Build and Upload wheel
44
- uses : pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main
45
+ uses : pytorch/test-infra/.github/workflows/build_wheels_linux.yml@release/2.8
45
46
with :
46
47
repository : pytorch/torchcodec
47
48
ref : " "
48
49
test-infra-repository : pytorch/test-infra
49
- test-infra-ref : main
50
+ test-infra-ref : release/2.8
50
51
build-matrix : ${{ needs.generate-matrix.outputs.matrix }}
51
52
pre-script : packaging/pre_build_script.sh
52
53
post-script : packaging/post_build_script.sh
@@ -66,13 +67,10 @@ jobs:
66
67
# PR.
67
68
# For the actual release we should add that label and change this to
68
69
# include more python versions.
69
- python-version : ['3.10']
70
- # We test against 12.6 to avoid having too big of a CI matrix,
71
- # but for releases we should add 12.8.
72
- # TODO add 13.0!
73
- cuda-version : ['12.6']
70
+ python-version : ['3.9', '3.10', '3.11', '3.12', '3.13']
71
+ cuda-version : ['12.6', '12.8', '12.9']
74
72
# TODO: put back ffmpeg 5 https://github.com/pytorch/torchcodec/issues/325
75
- ffmpeg-version-for-tests : ['4.4.2', '6', '7']
73
+ ffmpeg-version-for-tests : ['4.4.2', '5', ' 6', '7']
76
74
77
75
container :
78
76
image : " pytorch/manylinux2_28-builder:cuda${{ matrix.cuda-version }}"
90
88
name : pytorch_torchcodec__${{ matrix.python-version }}_cu${{ env.cuda_version_without_periods }}_x86_64
91
89
path : pytorch/torchcodec/dist/
92
90
- name : Setup miniconda using test-infra
93
- uses : pytorch/test-infra/.github/actions/setup-miniconda@main
91
+ uses : pytorch/test-infra/.github/actions/setup-miniconda@release/2.8
94
92
with :
95
93
python-version : ${{ matrix.python-version }}
96
94
# We install conda packages at the start because otherwise conda may have conflicts with dependencies.
@@ -109,7 +107,7 @@ jobs:
109
107
run : ${CONDA_RUN} python -m pip install --upgrade pip
110
108
- name : Install PyTorch
111
109
run : |
112
- ${CONDA_RUN} python -m pip install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly /cu${{ env.cuda_version_without_periods }}
110
+ ${CONDA_RUN} python -m pip install torch torchvision --index-url https://download.pytorch.org/whl/cu${{ env.cuda_version_without_periods }}
113
111
${CONDA_RUN} python -c 'import torch; print(f"{torch.__version__}"); print(f"{torch.__file__}"); print(f"{torch.cuda.is_available()=}")'
114
112
- name : Install torchcodec from the wheel
115
113
run : |
0 commit comments