-
Notifications
You must be signed in to change notification settings - Fork 84
60 lines (57 loc) · 2.05 KB
/
test_build_wheels_windows_with_cuda.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
name: Test Build Windows Wheels with CUDA
on:
pull_request:
paths:
- .github/actions/setup-binary-builds/action.yml
- .github/workflows/test_build_wheels_windows_with_cuda.yml
- .github/workflows/build_wheels_windows.yml
- .github/workflows/generate_binary_build_matrix.yml
- tools/scripts/generate_binary_build_matrix.py
workflow_dispatch:
permissions:
id-token: write
contents: read
jobs:
generate-matrix:
uses: ./.github/workflows/generate_binary_build_matrix.yml
with:
package-type: wheel
os: windows
test-infra-repository: ${{ github.repository }}
test-infra-ref: ${{ github.ref }}
test:
needs: generate-matrix
strategy:
fail-fast: false
matrix:
include:
- repository: pytorch/audio
env-script: packaging/vc_env_helper.bat
wheel-build-params: "--plat-name win_amd64"
smoke-test-script: test/smoke_test/smoke_test.py
package-name: torchaudio
- repository: pytorch/vision
pre-script: packaging/pre_build_script.sh
env-script: packaging/windows/internal/vc_env_helper.bat
post-script: "python packaging/wheel/relocate.py"
smoke-test-script: test/smoke_test.py
package-name: torchvision
cache-path: ""
cache-key: ""
uses: ./.github/workflows/build_wheels_windows.yml
name: ${{ matrix.repository }}
with:
repository: ${{ matrix.repository }}
ref: nightly
test-infra-repository: ${{ github.repository }}
test-infra-ref: ${{ github.ref }}
build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
pre-script: ${{ matrix.pre-script }}
env-script: ${{ matrix.env-script }}
wheel-build-params: ${{ matrix.wheel-build-params }}
post-script: ${{ matrix.post-script }}
smoke-test-script: ${{ matrix.smoke-test-script }}
package-name: ${{ matrix.package-name }}
trigger-event: "${{ github.event_name }}"
cache-path: ${{ matrix.cache-path }}
cache-key: ${{ matrix.cache-key }}