forked from IBMSparkGPU/GPUEnabler
-
Notifications
You must be signed in to change notification settings - Fork 0
/
compile.sh
executable file
·48 lines (38 loc) · 1.23 KB
/
compile.sh
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
#!/bin/bash
#
# Check for Maven
#
type mvn >/dev/null 2>&1 || {
echo "Maven is missing to compile the source files. Kindly install maven to proceed."
exit -1
}
export MAVEN_OPTS="-Xmx2048m -XX:MaxPermSize=1024m"
CUDA=0
CUDAVER=0
#
# Identify the CUDA runtime version to activate the correct profile
# Install Cuda for your platform from
# https://developer.nvidia.com/cuda-downloads.
#
type nvcc >/dev/null 2>&1 && CUDA=1
if [[ $CUDA != 0 ]]; then
CUDAVER=`nvcc --version | tail -1 | awk '{ print \$5 }' | cut -d ',' -f 1`
if [[ $CUDAVER == "7.5" ]]; then
echo "Identified CUDA version is 7.5"
CUDAVER="jcuda75"
elif [[ $CUDAVER == "7.0" ]]; then
echo "Identified CUDA version is 7.0a"
CUDAVER="jcuda70a"
else
echo "Not a supported version. Installation will fallback to default"
fi
fi
MVN_CMD="mvn"
if [[ $CUDAVER != 0 ]]; then
MVN_ARGS="-P$CUDAVER -Dmaven.compiler.showWarnings=true -Dmaven.compiler.showDeprecation=true"
else
MVN_ARGS="-Dmaven.compiler.showWarnings=true -Dmaven.compiler.showDeprecation=true"
fi
echo "Executing :: $MVN_CMD $MVN_ARGS -DskipTests $@ clean install "
$MVN_CMD $MVN_ARGS -DskipTests $@ clean install 2>&1 | tee ~/compile.txt
# ./utils/embed.sh -d gpu-enabler_2.11-1.0.0.jar