@@ -25,12 +25,12 @@ 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.7
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.7
34
34
with-cpu : disable
35
35
with-xpu : disable
36
36
with-rocm : disable
@@ -41,12 +41,12 @@ jobs:
41
41
strategy :
42
42
fail-fast : false
43
43
name : Build and Upload wheel
44
- uses : pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main
44
+ uses : pytorch/test-infra/.github/workflows/build_wheels_linux.yml@release/2.7
45
45
with :
46
46
repository : pytorch/torchcodec
47
47
ref : " "
48
48
test-infra-repository : pytorch/test-infra
49
- test-infra-ref : main
49
+ test-infra-ref : release/2.7
50
50
build-matrix : ${{ needs.generate-matrix.outputs.matrix }}
51
51
pre-script : packaging/pre_build_script.sh
52
52
post-script : packaging/post_build_script.sh
@@ -66,10 +66,10 @@ jobs:
66
66
# PR.
67
67
# For the actual release we should add that label and change this to
68
68
# include more python versions.
69
- python-version : ['3.9']
69
+ python-version : ['3.9', '3.10', '3.11', '3.12', '3.13' ]
70
70
cuda-version : ['11.8', '12.6', '12.8']
71
71
# TODO: put back ffmpeg 5 https://github.com/pytorch/torchcodec/issues/325
72
- ffmpeg-version-for-tests : ['4.4.2', '6', '7']
72
+ ffmpeg-version-for-tests : ['4.4.2', '5', ' 6', '7']
73
73
74
74
container :
75
75
image : " pytorch/manylinux2_28-builder:cuda${{ matrix.cuda-version }}"
87
87
name : pytorch_torchcodec__${{ matrix.python-version }}_cu${{ env.cuda_version_without_periods }}_x86_64
88
88
path : pytorch/torchcodec/dist/
89
89
- name : Setup miniconda using test-infra
90
- uses : pytorch/test-infra/.github/actions/setup-miniconda@main
90
+ uses : pytorch/test-infra/.github/actions/setup-miniconda@release/2.7
91
91
with :
92
92
python-version : ${{ matrix.python-version }}
93
93
# We install conda packages at the start because otherwise conda may have conflicts with dependencies.
@@ -105,7 +105,7 @@ jobs:
105
105
run : ${CONDA_RUN} python -m pip install --upgrade pip
106
106
- name : Install PyTorch
107
107
run : |
108
- ${CONDA_RUN} python -m pip install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly /cu${{ env.cuda_version_without_periods }}
108
+ ${CONDA_RUN} python -m pip install torch torchvision --index-url https://download.pytorch.org/whl/cu${{ env.cuda_version_without_periods }}
109
109
${CONDA_RUN} python -c 'import torch; print(f"{torch.__version__}"); print(f"{torch.__file__}"); print(f"{torch.cuda.is_available()=}")'
110
110
- name : Install torchcodec from the wheel
111
111
run : |
0 commit comments