22
22
- all
23
23
24
24
jobs :
25
- generate-windows-conda- matrix :
25
+ generate-windows-matrix :
26
26
uses : pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@main
27
27
with :
28
- package-type : conda
29
- os : windows
30
- channel : ${{ inputs.channel }}
31
- generate-windows-wheel-matrix :
32
- uses : pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@main
33
- with :
34
- package-type : wheel
35
- os : windows
36
- channel : ${{ inputs.channel }}
37
- generate-windows-libtorch-matrix :
38
- uses : pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@main
39
- with :
40
- package-type : libtorch
28
+ package-type : all
41
29
os : windows
42
30
channel : ${{ inputs.channel }}
43
31
44
- win-conda :
45
- needs : generate-windows-conda- matrix
32
+ win :
33
+ needs : generate-windows-matrix
46
34
strategy :
47
- matrix : ${{ fromJson(needs.generate-windows-conda- matrix.outputs.matrix) }}
35
+ matrix : ${{ fromJson(needs.generate-windows-matrix.outputs.matrix) }}
48
36
fail-fast : false
49
37
uses : pytorch/test-infra/.github/workflows/windows_job.yml@main
50
38
with :
@@ -57,29 +45,14 @@ jobs:
57
45
export GPU_ARCH_VER="${{ matrix.gpu_arch_version }}"
58
46
export GPU_ARCH_TYPE="${{ matrix.gpu_arch_type }}"
59
47
export INSTALLATION="${{ matrix.installation }}"
60
- export CUDA_VER="${{ inputs.desired_cuda }}"
61
- conda create -y -n ${ENV_NAME} python=${{ matrix.python_version }} numpy pillow
62
- conda activate ${ENV_NAME}
63
- eval $INSTALLATION
64
- python ./test/smoke_test/smoke_test.py
48
+ export CUDA_VER="${{ matrix.desired_cuda }}"
65
49
66
- win-wheel :
67
- needs : generate-windows-wheel-matrix
68
- uses : ./.github/workflows/validate-windows-binary.yml
69
- name : win-wheel
70
- with :
71
- test-matrix : ${{ needs.generate-windows-wheel-matrix.outputs.matrix }}
72
-
73
- win-libt :
74
- needs : generate-windows-libtorch-matrix
75
- strategy :
76
- matrix :
77
- ${{ fromJson(needs.generate-windows-libtorch-matrix.outputs.matrix) }}
78
- fail-fast : false
79
- uses : pytorch/test-infra/.github/workflows/windows_job.yml@main
80
- with :
81
- job-name : ${{ inputs.build_name }}
82
- script : |
83
- set -ex
84
- curl ${{ matrix.installation }} -o libtorch.zip
85
- unzip libtorch.zip
50
+ if [[ ${{ matrix.package_type }} == "libtorch" ]]; then
51
+ curl ${{ matrix.installation }} -o libtorch.zip
52
+ unzip libtorch.zip
53
+ else
54
+ conda create -y -n ${ENV_NAME} python=${{ matrix.python_version }} numpy pillow
55
+ conda activate ${ENV_NAME}
56
+ eval $INSTALLATION
57
+ python ./test/smoke_test/smoke_test.py
58
+ fi
0 commit comments