GPRat/examples/gprat_cpp/run_gprat_cpp.sh at main · SC-SGS/GPRat · GitHub

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
#!/bin/bash
# $1 cpu/gpu

################################################################################
set -e  # Exit immediately if a command exits with a non-zero status.
#set -x  # Print each command before executing it.

################################################################################
# Configurations
################################################################################

if [[ -z "$1" ]]; then
    echo "Input parameter is missing. Using default: Run computations on CPU"
elif [[ "$1" == "gpu" ]]; then
    use_gpu="--use_gpu"
elif [[ "$1" != "cpu" ]]; then
    echo "Please specify input parameter: cpu/gpu"
    exit 1
fi

# Select BLAS library
if [[ "$2" == "mkl" ]]
then
    USE_MKL=ON
else
    USE_MKL=OFF
fi

if command -v spack &> /dev/null; then
    echo "Spack command found, checking for environments..."
    # Get current hostname
    HOSTNAME=$(hostname -s)

    if [[ "$HOSTNAME" == "ipvs-epyc1" ]]; then
	# Check if the gprat_cpu_gcc environment exists
    	if spack env list | grep -q "gprat_cpu_gcc"; then
	   echo "Found gprat_cpu_gcc environment, activating it."
	    module load gcc/14.2.0
	    export CXX=g++
	    export CC=gcc
	    spack env activate gprat_cpu_gcc
	    GPRAT_WITH_CUDA=OFF # whether GPRAT_WITH_CUDA is ON of OFF is irrelevant for this example
	    GPRAT_APEX_STEPS=OFF
	    GPRAT_APEX_CHOLESKY=OFF
	fi
    elif [[ "$HOSTNAME" == "sven0"  ||  "$HOSTNAME" == "sven1" ]]; then
	#module load gcc/13.2.1
	spack load openblas arch=linux-fedora38-riscv64
	HPX_CMAKE=$HOME/git_workspace/build-scripts/build/hpx/lib64/cmake/HPX
        GPRAT_WITH_CUDA=OFF
	GPRAT_APEX_STEPS=OFF
	GPRAT_APEX_CHOLESKY=OFF
	ADD=64
    elif [[ $(uname -i) == "aarch64" ]]; then
	spack load [email protected]
	# Check if the gprat_cpu_arm environment exists
	if spack env list | grep -q "gprat_cpu_arm"; then
	    echo "Found gprat_cpu_arm environment, activating it."
	    spack env activate gprat_cpu_arm
	fi
	GPRAT_WITH_CUDA=OFF
	GPRAT_APEX_STEPS=OFF
	GPRAT_APEX_CHOLESKY=OFF
	ADD=64
    elif [[ "$HOSTNAME" == "simcl1n1" || "$HOSTNAME" == "simcl1n2" ]]; then
	# Check if the gprat_gpu_clang environment exists
	if spack env list | grep -q "gprat_gpu_clang"; then
	    echo "Found gprat_gpu_clang environment, activating it."
	    module load clang/17.0.1
	    export CXX=clang++
	    export CC=clang
	    module load cuda/12.0.1
	    spack env activate gprat_gpu_clang
	    GPRAT_WITH_CUDA=ON
	    GPRAT_APEX_STEPS=OFF
	    GPRAT_APEX_CHOLESKY=OFF
	fi
    else
    	echo "Hostname is $HOSTNAME — no action taken."
    fi
else
    echo "Spack command not found. Building example without Spack."
    # Assuming that Spack is not required on given system
fi

# Configure APEX
export APEX_SCREEN_OUTPUT=0
export APEX_DISABLE=1

################################################################################
# Compile code
################################################################################
rm -rf build && mkdir build && cd build

# Configure the project
cmake .. -DCMAKE_BUILD_TYPE=Release \
         -DGPRat_DIR=./lib$ADD/cmake/GPRat \
         -DGPRAT_WITH_CUDA=${GPRAT_WITH_CUDA} \
	 -DGPRAT_APEX_STEPS=${GPRAT_APEX_STEPS} \
	 -DGPRAT_APEX_CHOLESKY=${GPRAT_APEX_CHOLESKY} \
         -DHPX_DIR=$HPX_CMAKE \
	 -DUSE_MKL=$USE_MKL

# Build the project
make -j

################################################################################
# Run code
################################################################################

./gprat_cpp $use_gpu