mirror of https://github.com/alibaba/MNN.git
34 lines
1.0 KiB
PowerShell
34 lines
1.0 KiB
PowerShell
|
Param(
|
||
|
|
[Parameter(Mandatory=$true)][String]$path,
|
||
|
|
[Switch]$opencl
|
||
|
|
)
|
||
|
|
$erroractionpreference = "stop"
|
||
|
|
Remove-Item $path -Recurse -ErrorAction Ignore
|
||
|
|
mkdir -p $path
|
||
|
|
$TOOLS_PATH = $(Resolve-Path $path).Path
|
||
|
|
|
||
|
|
powershell ./schema/generate.ps1
|
||
|
|
|
||
|
|
$CMAKE_ARGS = "-DMNN_SEP_BUILD=OFF -DMNN_BUILD_TRAIN=ON -DMNN_BUILD_TOOLS=ON -DMNN_BUILD_QUANTOOLS=ON -DMNN_EVALUATION=ON -DMNN_BUILD_CONVERTER=ON -DMNN_BUILD_BENCHMARK=ON -DMNN_BUILD_TEST=ON"
|
||
|
|
if ($opencl) {
|
||
|
|
$CMAKE_ARGS = "$CMAKE_ARGS -DMNN_OPENCL=ON"
|
||
|
|
}
|
||
|
|
|
||
|
|
Remove-Item build -Recurse -ErrorAction Ignore
|
||
|
|
mkdir build
|
||
|
|
pushd build
|
||
|
|
|
||
|
|
Remove-Item CMakeCache.txt -ErrorAction Ignore
|
||
|
|
Invoke-Expression "cmake -G Ninja $CMAKE_ARGS -DCMAKE_BUILD_TYPE=Release -DMNN_WIN_RUNTIME_MT=ON -DMNN_BUILD_SHARED_LIBS=OFF .."
|
||
|
|
ninja
|
||
|
|
cp MNNV2Basic.out.exe $TOOLS_PATH
|
||
|
|
cp MNNConvert.exe $TOOLS_PATH
|
||
|
|
cp testModel.out.exe $TOOLS_PATH
|
||
|
|
cp run_test.out.exe $TOOLS_PATH
|
||
|
|
cp quantized.out.exe $TOOLS_PATH
|
||
|
|
cp train.out.exe $TOOLS_PATH
|
||
|
|
cp benchmark.out.exe $TOOLS_PATH
|
||
|
|
cp benchmarkExprModels.out.exe $TOOLS_PATH
|
||
|
|
cp backendTest.out.exe $TOOLS_PATH
|
||
|
|
|
||
|
|
popd
|