CNTK/Tools/make_binary_drop_windows.ps1

165 строки
5.1 KiB
PowerShell

# Copyright (c) Microsoft. All rights reserved.
# Licensed under the MIT license. See LICENSE file in the project root for full license information.
# WARNING. This will run in Microsoft Internal Environment ONLY
# Generating CNTK Binary drops in Jenkins environment
# Command line parameters
# Verbose command line parameter (-verbose) is automatically added
# because of CmdletBinding
[CmdletBinding()]
param
(
# Supposed to be taken from Jenkins BUILD_CONFIGURATION
[string]$buildConfig,
# Supposed to be taken from Jenkins TARGET_CONFIGURATION
[string]$targetConfig,
# File share path. Supposed to have sub-folders corresponding to $targetConfig
[string]$sharePath
)
# Set to Stop on Error
$ErrorActionPreference = 'Stop'
# Manual parameters check rather than using [Parameter(Mandatory=$True)]
# to avoid the risk of interactive prompts inside a Jenkins job
$usage = " parameter is missing. Usage example: make_binary_drop_windows.ps1 -buildConfig Release -targetConfig gpu -sharePath \\server\share"
If (-not $buildConfig) {Throw "buildConfig" + $usage}
If (-not $targetConfig) {Throw "targetConfig" + $usage}
If (-not $sharePath) {Throw "sharePath" + $usage}
# Set Verbose mode
If ($verbose)
{
$VerbosePreference = "continue"
}
Write-Verbose "Making binary drops..."
# If not a Release build quit
If ($buildConfig -ne "Release")
{
Write-Verbose "Not a release build. No binary drops generation"
Exit
}
# Set Paths
$basePath = "BinaryDrops\ToZip"
$baseDropPath = Join-Path $basePath -ChildPath cntk
$baseIncludePath = Join-Path $baseDropPath -ChildPath Include
$zipFile = "BinaryDrops\BinaryDrops.zip"
$buildPath = "x64\Release"
If ($targetConfig -eq "CPU")
{
$buildPath = "x64\Release_CpuOnly"
}
# Include Files
$includePath = "Source\Common\Include"
$includePath20 = "Source\CNTKv2LibraryDll\API"
$includeFiles = New-Object string[] 3
$includeFiles[0] = Join-Path $includePath -ChildPath Eval.h
$includeFiles[1] = Join-Path $includePath20 -ChildPath CNTKLibrary.h
$includeFiles[2] = Join-Path $includePath20 -ChildPath CNTKLibraryInternals.h
$sharePath = Join-Path $sharePath -ChildPath $targetConfig
# Make binary drop folder
New-Item -Path $baseDropPath -ItemType directory
# Copy build binaries
Write-Verbose "Copying build binaries ..."
Copy-Item $buildPath -Recurse -Destination $baseDropPath\cntk
# Clean unwanted items
Remove-Item $baseDropPath\cntk\*test*.exe
Remove-Item $baseDropPath\cntk\*.pdb
# Keep EvalDll.lib
Remove-Item $baseDropPath\cntk\*.lib -Exclude EvalDll.lib, CNTKLibrary-2.0.lib
Remove-Item $baseDropPath\cntk\*.exp
Remove-Item $baseDropPath\cntk\*.metagen
# Remove specific items
# Add Python artefacts to 2.0 Beta Drop
# If (Test-Path $baseDropPath\cntk\Python)
# {
# Remove-Item $baseDropPath\cntk\Python -Recurse
# }
# Add CNTKLibrary-2.0.dll to 2.0 Beta Drop
# If (Test-Path $baseDropPath\cntk\CNTKLibrary-2.0.dll)
# {
# Remove-Item $baseDropPath\cntk\CNTKLibrary-2.0.dll
# }
If (Test-Path $baseDropPath\cntk\CPPEvalClientTest.exe)
{
Remove-Item $baseDropPath\cntk\CPPEvalClientTest.exe
}
If (Test-Path $baseDropPath\cntk\CSEvalClientTest.exe)
{
Remove-Item $baseDropPath\cntk\CSEvalClientTest.exe
}
If (Test-Path $baseDropPath\cntk\CSEvalClientTest.exe.config)
{
Remove-Item $baseDropPath\cntk\CSEvalClientTest.exe.config
}
If (Test-Path $baseDropPath\cntk\CommandEval.exe)
{
Remove-Item $baseDropPath\cntk\CommandEval.exe
}
# Make Include folder
New-Item -Path $baseIncludePath -ItemType directory
# Copy Include
Write-Verbose "Copying Include files ..."
Foreach ($includeFile in $includeFiles)
{
Copy-Item $includeFile -Destination $baseIncludePath
}
# Copy Examples
Write-Verbose "Copying Examples ..."
Copy-Item Examples -Recurse -Destination $baseDropPath\Examples
# Include CPPEvalV2Client examples in 2.0 Beta drop
# If (Test-Path $baseDropPath\Examples\Evaluation\CPPEvalV2Client)
# {
# Remove-Item $baseDropPath\Examples\Evaluation\CPPEvalV2Client -Recurse
# }
# Copy Scripts
Write-Verbose "Copying Scripts ..."
Copy-Item Scripts -Recurse -Destination $baseDropPath\Scripts
# Remove test related file(s) if exist(s)
If (Test-Path $baseDropPath\Scripts\pytest.ini)
{
Remove-Item $baseDropPath\Scripts\pytest.ini
}
# Copy all items from the share
# For whatever reason Copy-Item in the line below does not work
# Copy-Item $sharePath"\*" -Recurse -Destination $baseDropPath
# Copying with Robocopy. Maximum 2 retries, 30 sec waiting time in between
Write-Verbose "Copying dependencies and other files from Remote Share ..."
robocopy $sharePath $baseDropPath /s /e /r:2 /w:30
# Check that Robocopy finished OK.
# Any exit code greater than 7 indicates error
# See http://ss64.com/nt/robocopy-exit.html
If ($LastExitCode -gt 7)
{
Throw "Copying from Remote Share failed. Robocopy exit code is " + $LastExitCode
}
Write-Verbose "Making ZIP and cleaning up..."
# Make ZIP file
$source = Join-Path $PWD.Path -ChildPath $basePath
$destination = Join-Path $PWD.Path -ChildPath $zipFile
Add-Type -assembly "system.io.compression.filesystem"
[io.compression.zipfile]::CreateFromDirectory($source, $destination)
# Remove ZIP sources
If (Test-Path $basePath)
{
Remove-Item $basePath -Recurse
}