|
|
|
@ -15,6 +15,16 @@
|
|
|
|
|
@echo off
|
|
|
|
|
@title mindspore_build
|
|
|
|
|
|
|
|
|
|
SET BASE_PATH=%CD%
|
|
|
|
|
SET BUILD_PATH=%BASE_PATH%/build
|
|
|
|
|
|
|
|
|
|
SET threads=6
|
|
|
|
|
SET X86_64_SIMD=off
|
|
|
|
|
|
|
|
|
|
set VERSION_MAJOR=''
|
|
|
|
|
set VERSION_MINOR=''
|
|
|
|
|
set ERSION_REVISION=''
|
|
|
|
|
|
|
|
|
|
find "const int ms_version_major =" mindspore\lite\include\version.h > version.txt
|
|
|
|
|
for /f "delims=\= tokens=2" %%a in ('findstr "const int ms_version_major = " version.txt') do (set x=%%a)
|
|
|
|
|
set VERSION_MAJOR=%x:~1,1%
|
|
|
|
@ -46,7 +56,7 @@ IF %errorlevel% == 0 (
|
|
|
|
|
SET X86_64_SIMD=avx512
|
|
|
|
|
) ELSE (
|
|
|
|
|
echo "MindSpore_lite the second parameter must in [avx, avx512, sse, off], but now is [%2%]"
|
|
|
|
|
call :run_fail
|
|
|
|
|
call :clean
|
|
|
|
|
)
|
|
|
|
|
IF NOT "%3%" == "" (
|
|
|
|
|
SET threads=%3%
|
|
|
|
@ -54,8 +64,6 @@ IF %errorlevel% == 0 (
|
|
|
|
|
)
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
SET BASE_PATH=%CD%
|
|
|
|
|
SET BUILD_PATH=%BASE_PATH%/build
|
|
|
|
|
IF NOT EXIST "%BUILD_PATH%" (
|
|
|
|
|
md "build"
|
|
|
|
|
)
|
|
|
|
@ -78,13 +86,13 @@ IF "%1%" == "lite" (
|
|
|
|
|
)
|
|
|
|
|
IF NOT %errorlevel% == 0 (
|
|
|
|
|
echo "cmake fail."
|
|
|
|
|
call :run_fail
|
|
|
|
|
call :clean
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
cmake --build . --target package -- -j%threads%
|
|
|
|
|
IF NOT %errorlevel% == 0 (
|
|
|
|
|
echo "build fail."
|
|
|
|
|
call :run_fail
|
|
|
|
|
call :clean
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
IF EXIST "%BASE_PATH%/output" (
|
|
|
|
@ -92,12 +100,11 @@ IF EXIST "%BASE_PATH%/output" (
|
|
|
|
|
rd /s /q _CPack_Packages
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
goto run_eof
|
|
|
|
|
cd %BASE_PATH%
|
|
|
|
|
set errorlevel=0
|
|
|
|
|
EXIT /b %errorlevel%
|
|
|
|
|
|
|
|
|
|
:run_fail
|
|
|
|
|
:clean
|
|
|
|
|
cd %BASE_PATH%
|
|
|
|
|
set errorlevel=1
|
|
|
|
|
EXIT /b %errorlevel%
|
|
|
|
|
|
|
|
|
|
:run_eof
|
|
|
|
|
cd %BASE_PATH%
|
|
|
|
|
EXIT /b %errorlevel%
|