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
|
#!/bin/bash
# Optional arguments. CUDA_VERSIONS must be supplied if CUDA_CC is specified.
# Example usage: ./deploy_linux.bash "12.6" "86 87 89"
DEFAULT_CUDA_VERSIONS=("10.0" "10.1" "10.2" "11.0" "12.0" "12.6" "12.9")
INPUT_CUDA_VERSIONS=(${1:-${DEFAULT_CUDA_VERSIONS[@]}})
INPUT_CUDA_CC="$2" # Optional string: "86 87 89"
# The cuda versions against which we will compile mumax3
for CUDAVERSION in "${INPUT_CUDA_VERSIONS[@]}"; do
#! NOTE: each CUDA version has a MAXIMUM GCC version: https://stackoverflow.com/a/46380601
#! EDIT IF-ELSE BELOW TO REFER TO YOUR INSTALLED GCC VERSION(S)!
if [ 1 -eq "$(echo "${CUDAVERSION} < 11.0" | bc)" ]; then
export NVCC_CCBIN=/usr/bin/gcc-7
else
export NVCC_CCBIN=/usr/bin/gcc
fi
# The final location of the mumax3 executables and libs
MUMAX3UNAME=mumax3.11.1_linux_cuda${CUDAVERSION}
BUILDDIR=./build/${MUMAX3UNAME}
rm -rf $BUILDDIR
mkdir -p $BUILDDIR
# The location of the home dirctory of this cuda version
# We export this variable so that cuda/Makefile knows how to build the wrappers
export CUDA_HOME=/usr/local/cuda-${CUDAVERSION}
# All supported compute capabilities of this cuda version
# See supported CC for each CUDA version at https://stackoverflow.com/a/28933055
# See min. driver version for each CUDA version at https://docs.nvidia.com/deploy/cuda-compatibility/#minor-version-compatibility
# We export CUDA_CC so that cuda/Makefile knows what to include in the fat wrappers
if [ -z "$INPUT_CUDA_CC" ]; then
case $CUDAVERSION in
"10.0") export CUDA_CC="50 52 53 60 61 62 70 72 75";; # Min. Linux driver: >=410.48
"10.1") export CUDA_CC="50 52 53 60 61 62 70 72 75";; # Min. Linux driver: >=418.39
"10.2") export CUDA_CC="50 52 53 60 61 62 70 72 75";; # Min. Linux driver: >=440.33
"11.0") export CUDA_CC="50 52 53 60 61 62 70 72 75 80";; # Min. Linux driver: >=450.80.02
"12.0") export CUDA_CC="50 52 53 60 61 62 70 72 75 80 86 87 89 90";; # Min. Linux driver: >=525.60.13 (same for all 12.x)
"12.6") export CUDA_CC="50 52 53 60 61 62 70 72 75 80 86 87 89 90";; # Highest CUDA version supporting CC < 7.5
"12.9") export CUDA_CC=" 75 80 86 87 89 90 100 120";;
esac
else
export CUDA_CC="$INPUT_CUDA_CC"
fi
# Print important variables (debug info)
echo "[INFO] CUDA Versions: ${INPUT_CUDA_VERSIONS[*]}"
echo "[INFO] Compute Capabilities: ${INPUT_CUDA_CC}"
echo "[INFO] GOPATH: ${GOPATH}"
# The path for shared libraries (relative to the build directory)
RPATH=lib
mkdir -p $BUILDDIR/$RPATH
# We overwrite the CGO Flags to make sure that it is compiled against $CUDAVERSION
export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH
export CGO_LDFLAGS="-lcufft -lcurand -lcuda -L${CUDA_HOME}/lib64 -Wl,-rpath -Wl,\$ORIGIN/$RPATH"
export CGO_CFLAGS="-I${CUDA_HOME}/include"
# (Re)build everything
(cd .. && make realclean && make -j 4 || exit 1)
# Copy the executable and the cuda libraries to the output directory
cp $GOPATH/bin/mumax3 $BUILDDIR
cp $GOPATH/bin/mumax3-convert $BUILDDIR
cp $GOPATH/bin/mumax3-server $BUILDDIR
cp ../LICENSE $BUILDDIR
cp $( ldd ${BUILDDIR}/mumax3 | grep libcufft | awk '{print $3}' ) ${BUILDDIR}/${RPATH}
cp $( ldd ${BUILDDIR}/mumax3 | grep libcurand | awk '{print $3}' ) ${BUILDDIR}/${RPATH}
(cd build && tar -czf ${MUMAX3UNAME}.tar.gz ${MUMAX3UNAME})
done
|
