Skip to content

Commit 6659a2e

Browse files
committed
Fix windows smoke test
1 parent 0b51c4f commit 6659a2e

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

windows/internal/smoke_test.bat

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ set "PATH=%CD%\Python%PYTHON_VERSION%\Scripts;%CD%\Python;%PATH%"
5151
pip install -q numpy protobuf "mkl>=2019"
5252
if errorlevel 1 exit /b 1
5353

54-
if "%TEST_NIGHTLY_PACKAGE%" == "1" (
54+
if "%PYTORCH_BUILD_VERSION%" == "1" (
5555
call internal\install_nightly_package.bat
5656
if errorlevel 1 exit /b 1
5757
) else (
@@ -106,7 +106,7 @@ set CUDA_VER_MAJOR=%CUDA_VERSION:~0,-1%
106106
set CUDA_VER_MINOR=%CUDA_VERSION:~-1,1%
107107
set CUDA_VERSION_STR=%CUDA_VER_MAJOR%.%CUDA_VER_MINOR%
108108

109-
if "%TEST_NIGHTLY_PACKAGE%" == "1" (
109+
if NOT "%PYTORCH_BUILD_VERSION%"=="%PYTORCH_BUILD_VERSION:dev=%" (
110110
call internal\install_nightly_package.bat
111111
if errorlevel 1 exit /b 1
112112
goto smoke_test
@@ -168,7 +168,7 @@ echo "install and test libtorch"
168168
if "%VC_YEAR%" == "2017" powershell internal\vs2017_install.ps1
169169
if ERRORLEVEL 1 exit /b 1
170170

171-
if "%TEST_NIGHTLY_PACKAGE%" == "1" (
171+
if NOT "%PYTORCH_BUILD_VERSION%"=="%PYTORCH_BUILD_VERSION:dev=%" (
172172
call internal\install_nightly_package.bat
173173
if errorlevel 1 exit /b 1
174174
) else (

0 commit comments

Comments
 (0)