Skip to content

Commit f4bec46

Browse files
committed
fix
1 parent 9482414 commit f4bec46

1 file changed

Lines changed: 13 additions & 2 deletions

File tree

.github/workflows/test_build_wheels_linux_without_cuda.yml

Lines changed: 13 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,17 @@ jobs:
2626
with-cuda: disable
2727
with-rocm: disable
2828
build-python-only: enable
29+
generate-matrix-torchvision:
30+
uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml
31+
with:
32+
package-type: wheel
33+
os: linux
34+
test-infra-repository: ${{ github.repository }}
35+
test-infra-ref: ${{ github.ref }}
36+
with-xpu: disable
37+
with-rocm: disable
38+
with-cuda: disable
39+
build-python-only: "disable"
2940
test-torchtune:
3041
needs: generate-matrix
3142
strategy:
@@ -49,7 +60,7 @@ jobs:
4960
torchvision
5061
torchao
5162
test-torchvision-extra-decoders:
52-
needs: generate-matrix
63+
needs: generate-matrix-torchvision
5364
strategy:
5465
fail-fast: false
5566
matrix:
@@ -66,7 +77,7 @@ jobs:
6677
ref: ""
6778
test-infra-repository: ${{ github.repository }}
6879
test-infra-ref: ${{ github.ref }}
69-
build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
80+
build-matrix: ${{ needs.generate-matrix-torchvision.outputs.matrix }}
7081
package-name: ${{ matrix.package-name }}
7182
pre-script: ${{ matrix.pre-script }}
7283
post-script: ${{ matrix.post-script }}

0 commit comments

Comments
 (0)