File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -138,9 +138,9 @@ jobs:
138138 run : |
139139 call C:\Users\nunchaku\miniconda3\condabin\activate.bat activate
140140 IF "${{ matrix.torch }}"=="pre" (
141- call scripts\build_windows_wheel_torch_nightly.cmd ${{ matrix.python }} ${{ matrix.torch }} ${{ matrix.cuda }}
141+ call scripts\build_windows_wheel_torch_nightly.cmd ${{ matrix.python }} ${{ matrix.torch }} ${{ matrix.cuda }} 16
142142 ) ELSE (
143- call scripts\build_windows_wheel.cmd ${{ matrix.python }} ${{ matrix.torch }} ${{ matrix.cuda }}
143+ call scripts\build_windows_wheel.cmd ${{ matrix.python }} ${{ matrix.torch }} ${{ matrix.cuda }} 16
144144 )
145145 - name : Upload wheels to GitHub Release
146146 uses : softprops/action-gh-release@v2
Original file line number Diff line number Diff line change @@ -123,9 +123,9 @@ jobs:
123123 run : |
124124 call C:\Users\nunchaku\miniconda3\condabin\activate.bat activate
125125 IF "${{ matrix.torch }}"=="pre" (
126- call scripts\build_windows_wheel_torch_nightly.cmd ${{ matrix.python }} ${{ matrix.torch }} ${{ matrix.cuda }}
126+ call scripts\build_windows_wheel_torch_nightly.cmd ${{ matrix.python }} ${{ matrix.torch }} ${{ matrix.cuda }} 16
127127 ) ELSE (
128- call scripts\build_windows_wheel.cmd ${{ matrix.python }} ${{ matrix.torch }} ${{ matrix.cuda }}
128+ call scripts\build_windows_wheel.cmd ${{ matrix.python }} ${{ matrix.torch }} ${{ matrix.cuda }} 16
129129 )
130130 - name : Upload wheels to GitHub Release
131131 uses : softprops/action-gh-release@v2
Original file line number Diff line number Diff line change @@ -5,9 +5,11 @@ setlocal enabledelayedexpansion
55set PYTHON_VERSION = %1
66set TORCH_VERSION = %2
77set CUDA_VERSION = %3
8+ set MAX_JOBS = %4
89
910set CUDA_SHORT_VERSION = %CUDA_VERSION:. =%
10- echo %CUDA_SHORT_VERSION%
11+ echo CUDA_SHORT_VERSION: %CUDA_SHORT_VERSION%
12+ echo MAX_JOBS: %MAX_JOBS%
1113
1214:: setup some variables
1315if " %TORCH_VERSION% " == " 2.8" (
Original file line number Diff line number Diff line change @@ -5,9 +5,11 @@ setlocal enabledelayedexpansion
55set PYTHON_VERSION = %1
66set TORCH_VERSION = %2
77set CUDA_VERSION = %3
8+ set MAX_JOBS = %4
89
910set CUDA_SHORT_VERSION = %CUDA_VERSION:. =%
10- echo %CUDA_SHORT_VERSION%
11+ echo CUDA_SHORT_VERSION: %CUDA_SHORT_VERSION%
12+ echo MAX_JOBS: %MAX_JOBS%
1113
1214:: conda environment name
1315set ENV_NAME = build_env_%PYTHON_VERSION% _%TORCH_VERSION%
You can’t perform that action at this time.
0 commit comments