forked from microsoft/CNTK
-
Notifications
You must be signed in to change notification settings - Fork 0
/
generate_build_info
executable file
·150 lines (136 loc) · 4.11 KB
/
generate_build_info
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
#!/bin/bash
# Description: this script is used to generated buildinfo.h in Source/CNTK
# which will contain the following infomation to be displayed at runtime:
# BUILDTYPE (release/debug)
# BUILDTYPE (GPU/CPU-only)
# WITH_1BITSGD (whether 1bit-SGD support was enabled)
# MATHLIB (MKL/ACML)
# CUDA_PATH (if exists, i.e., for GPU builds)
# CUB_PATH (if exists, i.e., for GPU builds)
# CUDNN_PATH (if exists, i.e., only for GPU builds)
# GIT_COMMIT (SHA1, and whether working directory is modified)
# GTT_BRANCH (the current branch)
# BUILDER (user under which the build was done)
# BUILDMACHINE (build machine)
# BUILDPATH (build path)
usage ()
{
echo "usage: $0 <Config.make>"
echo "-------------------------------------------------------------------"
echo "This script is used to generate buildinfo.h in Source/CNTK"
echo "This script needs to be called from the top level directory of CNTK project"
echo "This script assumes git can be used"
echo "This script assumes Config.make has been made"
echo "-------------------------------------------------------------------"
if [ ! -z "$1" ] ; then
echo "ERROR message: $1"
fi
exit 1
}
Has_Git()
{
if hash git 2>/dev/null; then
return 0
else
return 1
fi
}
makebuildinfo()
{
target=$1
BUILDTYPE=$2
MATHLIB=$3
GIT_COMMIT=$4
GIT_BRANCH=$5
CUDA_PATH=$6
CUB_PATH=$7
WITH_1BITSGD=$8
BUILDER=$9
BUILDMACHINE=$10
BUILDPATH=$11
(
printf "#ifndef _BUILDINFO_H\n"
printf "#define _BUILDINFO_H\n"
printf "#define _GIT_EXIST\n"
printf "#define _MATHLIB_ \"%s\"\n" "$MATHLIB"
printf "#define _BUILDSHA1_ \"%s\"\n" "$GIT_COMMIT"
printf "#define _BUILDBRANCH_ \"%s\"\n" "$GIT_BRANCH"
if [ -z "$CUDA_PATH" ]; then
printf "#define _BUILDTARGET_ \"CPU-only\"\n"
else
printf "#define _BUILDTARGET_ \"GPU\"\n"
printf "#define _CUDA_PATH_ \"%s\"\n" "$CUDA_PATH"
fi
if [ ! -z "$CUB_PATH" ]; then
printf "#define _CUB_PATH_ \"%s\"\n" "$CUB_PATH"
fi
if [ ! -z "$CUDNN_PATH" ]; then
printf "#define _CUDNN_PATH_ \"%s\"\n" $CUDNN_PATH
fi
printf "#define _BUILDTYPE_ \"%s\"\n" "$BUILDTYPE"
if [ ! -z "$WITH_1BITSGD" ]; then
printf "#define _WITH_1BITSGD_ \"yes\"\n"
else
printf "#define _WITH_1BITSGD_ \"no\"\n"
fi
printf "#define _BUILDER_ \"%s\"\n" "$BUILDER"
printf "#define _BUILDMACHINE_ \"%s\"\n" "$BUILDMACHINE"
printf "#define _BUILDPATH_ \"%s\"\n" "$BUILDPATH"
printf "#endif\n"
) > "$target"
}
#//////////////////////////////////////////////////////#
# main function #
#//////////////////////////////////////////////////////#
if [ $# -ne 1 ]; then
usage
fi
config=$1
# 1. check whether we have git and what is the sha-1 value
if Has_Git; then has_git=1; else has_git=0; usage "git does not exist"; fi
GIT_STATUS=' (modified)'
git diff --quiet && git diff --cached --quiet && GIT_STATUS=''
GIT_COMMIT=`git rev-parse HEAD`$GIT_STATUS
GIT_BRANCH=`git rev-parse --abbrev-ref HEAD`
# 2. looking into Config.make
if [ ! -e $config ] ; then
usage "Config.make not exists"
fi
source $config
# 3. whether we have CUDA_PATH
if [ -z "${CUDA_PATH+x}" ]; then
CUDAPATH=""
else
CUDAPATH=$CUDA_PATH
fi
# 4. whether we have CUB_PATH
if [ -z "${CUB_PATH+x}" ]; then
CUBPATH=""
else
CUBPATH=$CUB_PATH
fi
# 5. Build machine info
BUILDER=$USER
BUILDMACHINE=`hostname`
BUILDPATH=`pwd`
# 6. make buildinfo.h
# TODO should avoid rebuild if not changing
target=Source/CNTK/buildinfo.h
if [ ! -d Source ] ; then
usage
fi
if [ -e Source/CNTK/buildinfo.h ] ; then
rm Source/CNTK/buildinfo.h
fi
makebuildinfo \
"$target" \
"$BUILDTYPE" \
"$MATHLIB" \
"$GIT_COMMIT" \
"$GIT_BRANCH" \
"$CUDAPATH" \
"$CUBPATH" \
"$CNTK_ENABLE_1BitSGD" \
"$BUILDER" \
"$BUILDMACHINE" \
"$BUILDPATH"