Skip to content

Commit

Permalink
Binary drop script. Linux. Addressed Code Review comments
Browse files Browse the repository at this point in the history
  • Loading branch information
alexeyo26 authored and thhoens committed Mar 21, 2016
1 parent fce4f7b commit 2523f94
Show file tree
Hide file tree
Showing 2 changed files with 70 additions and 66 deletions.
1 change: 1 addition & 0 deletions .gitattributes
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,7 @@ configure text eol=lf
generate_build_info text eol=lf
run-test text eol=lf
run-test-common text eol=lf
make_binary_drop_linux text eol=lf

Makefile text
*.sln text
Expand Down
135 changes: 69 additions & 66 deletions Tools/make_binary_drop_linux
Original file line number Diff line number Diff line change
Expand Up @@ -16,64 +16,69 @@ verbose=false

# Process Command line parameters

usage="Usage: make_binary_drop_linux -b build_configuration -t target_configuration [-v]\n-v Verbose output\nExample: make_binary_drop_linux -b Release -t gpu -v"
usage="Usage: make_binary_drop_linux -b build_configuration -t target_configuration [-v] [-h]\nBuild and Target configurations should correspond to Jenkins environment\n-v Verbose output\n-h Displays this message\nExample: make_binary_drop_linux -b Release -t gpu -v"

if [[ $# == 0 ]]; then
echo "Command line parameters are missing" >&2
echo -e $usage >&2
exit 1
echo "Command line parameters are missing" >&2
echo -e $usage >&2
exit 1
fi

while getopts ":b::t::v" opt; do
case $opt in
b)
# Supposed to be taken from Jenkins BUILD_CONFIGURATION
buildConfig=$OPTARG
;;
t)
# Supposed to be taken from Jenkins TARGET_CONFIGURATION
targetConfig=$OPTARG
;;
v)
# Enables verbose output
verbose=true
;;
\?)
echo "Invalid parameter: -$OPTARG" >&2
echo -e $usage >&2
exit 1
;;
:)
echo "Paremeter -$OPTARG requires an argument." >&2
echo -e $usage >&2
exit 1
;;
esac
case $opt in
b)
# Supposed to be taken from Jenkins BUILD_CONFIGURATION
buildConfig=$OPTARG
;;
t)
# Supposed to be taken from Jenkins TARGET_CONFIGURATION
targetConfig=$OPTARG
;;
v)
# Enables verbose output
verbose=true
;;
h)
# Display Usage message and exit
echo -e $usage
exit 0
;;
\?)
echo "Invalid parameter: -$OPTARG" >&2
echo -e $usage >&2
exit 1
;;
:)
echo "Parameter -$OPTARG requires an argument." >&2
echo -e $usage >&2
exit 1
;;
esac
done

# Check mandatory parameters presence
if [[ -z "$buildConfig" ]]; then
echo "Parameter -b is missing." >&2
echo -e $usage >&2
exit 1
echo "Parameter -b is missing." >&2
echo -e $usage >&2
exit 1
fi

if [[ -z "$targetConfig" ]]; then
echo "Parameter -t is missing." >&2
echo -e $usage >&2
exit 1
echo "Parameter -t is missing." >&2
echo -e $usage >&2
exit 1
fi

# Make buildConfig and targetConfig lowercase
buildConfig=$(echo $buildConfig | tr '[:upper:]' '[:lower:]')
targetConfig=$(echo $targetConfig | tr '[:upper:]' '[:lower:]')
buildConfig=$(echo ${buildConfig} | tr '[:upper:]' '[:lower:]')
targetConfig=$(echo ${targetConfig} | tr '[:upper:]' '[:lower:]')

# Enable "verbose" mode if needed
# stderr is NOT changed
if [[ "$verbose" == true ]]; then
exec 3>&1
exec 3>&1
else
exec 3>/dev/null
exec 3>/dev/null
fi

# Define helper function
Expand All @@ -82,22 +87,21 @@ fi
# usage: CopyFilesFromList source_path file_name_array destination_path
function CopyFilesFromList ()
{
declare -a fileNames=(${!2})
for fileName in "${fileNames[@]}"
do
cp "$1"/"$fileName" "$3"
done
declare -a fileNames=(${!2})
for fileName in "${fileNames[@]}"
do
cp "$1/$fileName" "$3"
done
}

# Main script


echo "Making binary drops..." >&3

# If not a Release build quit
if [[ $buildConfig != "release" ]]; then
echo "Not a release build. No binary drops generaion" >&3
exit 0
echo "Not a release build. No binary drops generation" >&3
exit 0
fi

# Dependency files
Expand All @@ -106,10 +110,9 @@ fi
declare -a acmlFiles=("libacml_mp.so" "libiomp5.so")

# Open CV
declare -a opencvFiles=("libopencv_core.so.3.0" "libopencv_imgproc.so.3.0"
"libopencv_imgproc.so.3.0" "libopencv_imgcodecs.so.3.0")
declare -a opencvFiles=("libopencv_core.so.3.0" "libopencv_imgproc.so.3.0" "libopencv_imgproc.so.3.0" "libopencv_imgcodecs.so.3.0")

# CUDA
# CUDA
declare -a cudaFiles=("libcudart.so.7.0" "libcublas.so.7.0" "libcurand.so.7.0" "libcusparse.so.7.0")

# cuDNN
Expand All @@ -122,19 +125,19 @@ cudaPath="/usr/local/cuda/lib64"
cudnnPath="/usr/local/cudnn-4.0/cuda/lib64"

# Set build paths
buildPath="build/"$targetConfig"/release"
buildPath="build/$targetConfig/release"
basePath="BinaryDrops"
baseDropPath=$basePath"/cntk"
baseBinariesPath=$baseDropPath"/cntk"
baseDependenciesPath=$baseBinariesPath"/dependencies/lib"
extrasPath="Tools/cntk-binary-drop/linux/"$targetConfig
baseDropPath="$basePath/cntk"
baseBinariesPath="$baseDropPath/cntk"
baseDependenciesPath="$baseBinariesPath/dependencies/lib"
extrasPath="Tools/cntk-binary-drop/linux/$targetConfig"
gzipFile="BinaryDrops.tar.gz"

# Make BinaryDrops directory
mkdir -p $baseBinariesPath

# Copy build binaries
echo "Copying build binaries..." >&3
echo "Copying build binaries..." >&3
cp -r $buildPath/* $baseBinariesPath

# Copy Examples
Expand Down Expand Up @@ -162,24 +165,24 @@ CopyFilesFromList $opencvPath opencvFiles[@] $baseDependenciesPath
# GPU Drops only
if [[ $targetConfig != "cpu" ]]; then

# Copy CUDA
echo "Copying CUDA..." >&3
CopyFilesFromList $cudaPath cudaFiles[@] $baseDependenciesPath
# Copy CUDA
echo "Copying CUDA..." >&3
CopyFilesFromList $cudaPath cudaFiles[@] $baseDependenciesPath

# Copy cuDNN
echo "Copying cuDNN..." >&3
CopyFilesFromList $cudnnPath cudnnFiles[@] $baseDependenciesPath
# Copy cuDNN
echo "Copying cuDNN..." >&3
CopyFilesFromList $cudnnPath cudnnFiles[@] $baseDependenciesPath

fi

echo "Making Archive and cleaning up..." >&3
# Make GZipped TAR
cd $basePath
if [[ "$verbose" == true ]]; then
tar -cvzf $gzipFile cntk
else
tar -czf $gzipFile cntk
fi
tar -cvzf $gzipFile cntk
else
tar -czf $gzipFile cntk
fi

# Remove TAR sources
rm -r cntk

0 comments on commit 2523f94

Please sign in to comment.