forked from mindspore-Ecosystem/mindspore
windows CI support benchmark
This commit is contained in:
parent
9593e82e56
commit
3b862a5cda
|
@ -0,0 +1,95 @@
|
|||
@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 ============================================================================
|
||||
@echo off
|
||||
@title run_benchmark_nets
|
||||
|
||||
SET BASEPATH=%CD%
|
||||
SET TOOL_PATH=%1
|
||||
SET TOOL_PATH=%TOOL_PATH:"=%/windows_x64
|
||||
SET MODEL_PATH=%2
|
||||
SET MODEL_PATH=%MODEL_PATH:"=%/models/hiai
|
||||
SET BENCHMARK_BASE=%BASEPATH:"=%/output/benchmark
|
||||
|
||||
cd /d %BASEPATH%
|
||||
|
||||
IF EXIST "%BASEPATH%/output" (
|
||||
rd /s /q output
|
||||
)
|
||||
md output
|
||||
|
||||
SET OUTPUT_PATH=%BASEPATH%/output
|
||||
|
||||
cd /d %OUTPUT_PATH%
|
||||
|
||||
IF EXIST benchmark (
|
||||
rd /s /q benchmark
|
||||
)
|
||||
md benchmark
|
||||
SET ret_code=0
|
||||
7z x -r "%TOOL_PATH%/mindspore-lite-*-converter-win-cpu.zip" -o"%BENCHMARK_BASE%"
|
||||
IF NOT %errorlevel% == 0 (
|
||||
echo "Decompression of converter tool fail!"
|
||||
SET ret_code=1
|
||||
)
|
||||
IF %ret_code% == 0 (
|
||||
7z x -r "%TOOL_PATH%/mindspore-lite-*-win-runtime-x86-cpu.zip" -o"%BENCHMARK_BASE%"
|
||||
IF NOT %errorlevel% == 0 (
|
||||
echo "Decompression of runtime tool fail!"
|
||||
SET ret_code=1
|
||||
)
|
||||
)
|
||||
cd benchmark
|
||||
md ms
|
||||
cd mindspore-lite-*-converter-win-cpu/converter
|
||||
IF %ret_code% == 0 (
|
||||
converter_lite --outputFile="%BENCHMARK_BASE%/ms/scan_hms_angle1" --modelFile="%MODEL_PATH%/scan_hms_angle1.tflite" --fmk=TFLITE
|
||||
IF NOT %errorlevel% == 0 (
|
||||
echo "Model conversion of scan_hms_angle1.tflite fail!"
|
||||
SET ret_code=1
|
||||
)
|
||||
)
|
||||
|
||||
IF %ret_code% == 0 (
|
||||
converter_lite --outputFile="%BENCHMARK_BASE%/ms/scan_hms_detect" --modelFile="%MODEL_PATH%/scan_hms_detect.tflite" --fmk=TFLITE
|
||||
IF NOT %errorlevel% == 0 (
|
||||
echo "Model conversion of scan_hms_detect.tflite fail!"
|
||||
SET ret_code=1
|
||||
)
|
||||
)
|
||||
|
||||
cd /d %BENCHMARK_BASE%/mindspore-lite-*-win-runtime-x86-cpu/benchmark
|
||||
SET INPUT_BASE=%MODEL_PATH%/input_output/input
|
||||
SET OUTPUT_BASE=%MODEL_PATH%/input_output/output
|
||||
IF %ret_code% == 0 (
|
||||
benchmark --modelFile="%BENCHMARK_BASE%/ms/scan_hms_angle1.ms" --inDataFile="%INPUT_BASE%/scan_hms_angle1.tflite.ms.bin" --benchmarkDataFile="%OUTPUT_BASE%/scan_hms_angle1.tflite.ms.out"
|
||||
IF NOT %errorlevel% == 0 (
|
||||
echo "benchmark scan_hms_angle1 fail!"
|
||||
SET ret_code=1
|
||||
)
|
||||
)
|
||||
|
||||
IF %ret_code% == 0 (
|
||||
benchmark --modelFile="%BENCHMARK_BASE%/ms/scan_hms_detect.ms" --inDataFile="%INPUT_BASE%/scan_hms_detect.tflite.ms.bin" --benchmarkDataFile="%OUTPUT_BASE%/scan_hms_detect.tflite.ms.out"
|
||||
IF NOT %errorlevel% == 0 (
|
||||
echo "benchmark scan_hms_detect fail!"
|
||||
SET ret_code=1
|
||||
)
|
||||
)
|
||||
|
||||
cd /d %BASEPATH%
|
||||
|
||||
IF %ret_code% == 1 (
|
||||
SET errorlevel=1
|
||||
)
|
|
@ -0,0 +1,28 @@
|
|||
@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 ============================================================================
|
||||
@echo off
|
||||
@title win_run_test
|
||||
|
||||
SET BASEPATH=%CD%
|
||||
|
||||
call run_benchmark_nets.bat %1 %2
|
||||
IF NOT %errorlevel% == 0 (
|
||||
echo "benchmark fail!"
|
||||
SET errorlevel=1
|
||||
) ELSE (
|
||||
echo "run benchmark tests success."
|
||||
)
|
||||
|
||||
cd /d %BASEPATH%
|
Loading…
Reference in New Issue