未验证 提交 a1b6ae26 编写于 作者: S Sing_chan 提交者: GitHub

make whl-build task only reuse local third_party cache (#35858)

上级 8a239ae5
...@@ -76,6 +76,7 @@ if not defined NIGHTLY_MODE set PRECISION_TEST=OFF ...@@ -76,6 +76,7 @@ if not defined NIGHTLY_MODE set PRECISION_TEST=OFF
if not defined retry_times set retry_times=3 if not defined retry_times set retry_times=3
if not defined PYTHON_ROOT set PYTHON_ROOT=C:\Python37 if not defined PYTHON_ROOT set PYTHON_ROOT=C:\Python37
if not defined BUILD_DIR set BUILD_DIR=build if not defined BUILD_DIR set BUILD_DIR=build
set task_name=%1
rem ------initialize the python environment------ rem ------initialize the python environment------
set PYTHON_EXECUTABLE=%PYTHON_ROOT%\python.exe set PYTHON_EXECUTABLE=%PYTHON_ROOT%\python.exe
...@@ -363,14 +364,22 @@ echo echo ${md5_content}^>md5.txt >> cache.sh ...@@ -363,14 +364,22 @@ echo echo ${md5_content}^>md5.txt >> cache.sh
%cache_dir%\tools\busybox64.exe bash cache.sh %cache_dir%\tools\busybox64.exe bash cache.sh
set /p md5=< md5.txt set /p md5=< md5.txt
echo %task_name%|findstr build >nul && (
set THIRD_PARTY_HOME=%cache_dir:\=/%/third_party
set THIRD_PARTY_PATH=!THIRD_PARTY_HOME!/%md5%
echo %task_name% is a whl-build task, will only reuse local third_party cache.
goto :cmake_impl
) || (
echo %task_name% is a PR-CI-Windows task, will try to reuse bos and local third_party cache both.
)
if "%WITH_GPU%"=="ON" ( if "%WITH_GPU%"=="ON" (
for /F "delims=" %%# in ('nvcc --version^|findstr V1') do set cuda_version=%%# for /F %%# in ('dir /b /d "C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\"') do set cuda_version=%%#
set cuda_version=!cuda_version:~-7,4! set cuda_version=!cuda_version:~-4!
set sub_dir=cuda!cuda_version:.=! set sub_dir=cuda!cuda_version:.=!
) else ( ) else (
set sub_dir=cpu set sub_dir=cpu
) )
set THIRD_PARTY_HOME=%cache_dir:\=/%/third_party/%sub_dir% set THIRD_PARTY_HOME=%cache_dir:\=/%/third_party/%sub_dir%
set THIRD_PARTY_PATH=%THIRD_PARTY_HOME%/%md5% set THIRD_PARTY_PATH=%THIRD_PARTY_HOME%/%md5%
set UPLOAD_TP_FILE=OFF set UPLOAD_TP_FILE=OFF
...@@ -397,7 +406,7 @@ if not exist %THIRD_PARTY_PATH% ( ...@@ -397,7 +406,7 @@ if not exist %THIRD_PARTY_PATH% (
if not exist %THIRD_PARTY_PATH% ( set UPLOAD_TP_FILE=ON ) if not exist %THIRD_PARTY_PATH% ( set UPLOAD_TP_FILE=ON )
cd %work_dir%\%BUILD_DIR% cd %work_dir%\%BUILD_DIR%
) else ( ) else (
echo Found reusable third_party cache locally, will reuse it. echo Found reusable third_party cache in %THIRD_PARTY_PATH%, will reuse it.
) )
:cmake_impl :cmake_impl
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册