mindspore/build.bat

127 lines
3.6 KiB
Batchfile
Raw Normal View History

2020-04-18 09:41:25 +08:00
@rem Copyright 2020 Huawei Technologies Co., Ltd
@rem
@rem Licensed under the Apache License, Version 2.0 (the "License");
@rem you may not use this file except in compliance with the License.
@rem You may obtain a copy of the License at
@rem
@rem http://www.apache.org/licenses/LICENSE-2.0
@rem
@rem Unless required by applicable law or agreed to in writing, software
@rem distributed under the License is distributed on an "AS IS" BASIS,
@rem WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
@rem See the License for the specific language governing permissions and
@rem limitations under the License.
@rem ============================================================================
2020-04-15 17:36:48 +08:00
@echo off
@title mindspore_build
2020-04-27 11:24:13 +08:00
2020-04-15 17:36:48 +08:00
SET BASEPATH=%CD%
IF NOT EXIST "%BASEPATH%/build" (
md "build"
)
2020-04-27 11:24:13 +08:00
2020-08-29 14:04:50 +08:00
cd %BASEPATH%/build
set BUILD_PATH=%CD%
2020-04-27 11:24:13 +08:00
IF NOT EXIST "%BUILD_PATH%/mindspore" (
md "mindspore"
)
2020-04-27 11:24:13 +08:00
2020-08-29 14:04:50 +08:00
cd %CD%/mindspore
2020-04-27 11:24:13 +08:00
IF "%1%" == "lite" (
call :run_cmake
IF errorlevel 1 (
echo "cmake fail one time."
call :gene_protobuf
call :gene_flatbuffer
call :run_cmake
IF errorlevel 1 (
echo "cmake fail."
goto run_fail
)
) ELSE (
call :gene_protobuf
call :gene_flatbuffer
2020-04-15 17:36:48 +08:00
)
2020-04-27 11:24:13 +08:00
cd %BUILD_PATH%/mindspore
IF "%2%" == "" (
cmake --build . --target package -- -j6
) ELSE (
cmake --build . --target package -- -j%2%
)
IF errorlevel 1 (
echo "build fail."
goto run_fail
) ELSE (
2020-08-29 14:04:50 +08:00
cd %BASEPATH%/output
rd /s /q _CPack_Packages
)
) ELSE (
cmake -DCMAKE_BUILD_TYPE=Release -DENABLE_CPU=ON -DENABLE_MINDDATA=ON -DUSE_GLOG=ON ^
-G "CodeBlocks - MinGW Makefiles" ../..
IF NOT %errorlevel% == 0 (
echo "cmake fail."
goto run_fail
)
IF "%1%" == "" (
cmake --build . --target package -- -j6
2020-04-16 14:18:25 +08:00
) ELSE (
cmake --build . --target package -- -j%1%
2020-04-16 14:18:25 +08:00
)
IF NOT %errorlevel% == 0 (
echo "build fail."
goto run_fail
2020-04-15 17:36:48 +08:00
)
)
2020-04-15 17:36:48 +08:00
2020-08-29 14:04:50 +08:00
cd %BASEPATH%
2020-04-15 17:36:48 +08:00
goto run_eof
:run_cmake
2020-08-29 14:04:50 +08:00
cd %BUILD_PATH%/mindspore
cmake -DBUILD_DEVICE=on -DBUILD_CONVERTER=on -DPLATFORM_ARM64=off -DSUPPORT_TRAIN=off ^
-DCMAKE_BUILD_TYPE=Release -DSUPPORT_GPU=off -DBUILD_MINDDATA=off -DOFFLINE_COMPILE=off ^
-G "CodeBlocks - MinGW Makefiles" "%BASEPATH%/mindspore/lite"
GOTO:EOF
:gene_protobuf
IF NOT DEFINED MSLIBS_CACHE_PATH (
cd /d %BASEPATH%/build/mindspore/_deps/protobuf-src/_build
) ELSE (
cd /d %MSLIBS_CACHE_PATH%/protobuf_*/bin
)
SET PROTO_SRC_DIR=%BASEPATH%/mindspore/lite/tools/converter/parser/caffe
protoc "%PROTO_SRC_DIR%/*.proto" --proto_path="%PROTO_SRC_DIR%" --cpp_out="%PROTO_SRC_DIR%"
SET PROTO_SRC_DIR=%BASEPATH%/mindspore/lite/tools/converter/parser/onnx
protoc "%PROTO_SRC_DIR%/*.proto" --proto_path="%PROTO_SRC_DIR%" --cpp_out="%PROTO_SRC_DIR%"
cd /d %BUILD_PATH%/mindspore
GOTO:EOF
:gene_flatbuffer
IF NOT DEFINED MSLIBS_CACHE_PATH (
cd /d %BASEPATH%/build/mindspore/_deps/flatbuffers-src/_build
) ELSE (
cd /d %MSLIBS_CACHE_PATH%/flatbuffers_*/bin
)
SET FLAT_DIR=%BASEPATH%/mindspore/lite/schema
flatc -c -b -o "%FLAT_DIR%" "%FLAT_DIR%/*.fbs"
flatc -c -b --reflect-types --gen-mutable --reflect-names --gen-object-api -o "%FLAT_DIR%/inner" "%FLAT_DIR%/*.fbs"
2020-08-29 14:04:50 +08:00
SET FLAT_DIR=%BASEPATH%/mindspore/lite/tools/converter/parser/tflite
flatc -c -b --reflect-types --gen-mutable --reflect-names --gen-object-api -o "%FLAT_DIR%" "%FLAT_DIR%/*.fbs"
cd /d %BUILD_PATH%/mindspore
GOTO:EOF
2020-04-15 17:36:48 +08:00
:run_fail
2020-08-29 14:04:50 +08:00
cd %BASEPATH%
set errorlevel=1
2020-04-15 17:36:48 +08:00
:run_eof