added query for cuda compute capability

jobscript/base_config.sh

@@ -20,6 +20,7 @@ USE_CUDA="${USE_CUDA}"
 USE_HIP="${USE_HIP}"
 USE_OMP="${USE_OMP}"
 VISABLE_DEVICE="${VISABLE_DEVICE:-"0"}"
+CUDA_COMPUTE_CAP="OFF"
 
 BENCHMARKS_TO_BUILD="${BENCHMARKS_TO_BUILD}"
 
@@ -48,7 +49,11 @@ module load likwid/5.3.0
 module load cmake
 module load python
 
-[ "${USE_CUDA}" = "ON" ] && module load cuda
+if [ "${USE_CUDA}" = "ON" ];
+then
+    module load cuda
+    CUDA_COMPUTE_CAP=$(nvidia-smi --query-gpu=compute_cap --format=csv,noheader | uniq | tr -d '.' | tr '\n' ';' | sed 's/;$//')
+fi
 [ "${USE_MPI}" = "ON" ] && module load openmpi/4.1.1-gcc9.3-legacy
 
 module list
@@ -65,7 +70,6 @@ BASE_DIR="/scratch"
 SOURCE_DIR="walberla_ci"
 FULL_PATH="${BASE_DIR}/${SOURCE_DIR}"
 BUILD_DIR="${FULL_PATH}/build"
-# PYTHON_VENV="${WORK}/walberla_venv"
 VENV_NAME="walberla_venv"
 
 if { conda env list | grep "$VENV_NAME"; } >/dev/null 2>&1; then

jobscript/build.sh

@@ -14,15 +14,6 @@ git checkout $WALBERLA_BRANCH
 [[ -n "$WALBERLA_COMMIT" ]] && git checkout $WALBERLA_COMMIT
 git pull
 
-# PYTHON Setup
-# if [ ! -d "${PYTHON_VENV}" ];
-# then
-#     python3 -m venv $PYTHON_VENV
-# fi
-# source $PYTHON_VENV/bin/activate
-# pip install --upgrade pip setuptools wheel jinja2 py-cpuinfo lbmpy pystencils
-# pip list
-
 mkdir -p ${BUILD_DIR}
 cd ${BUILD_DIR}
 
@@ -50,6 +41,7 @@ WALBERLA_BUILD_WITH_MPI="${USE_MPI}"
 WALBERLA_BUILD_WITH_CUDA="${USE_CUDA}"
 WALBERLA_BUILD_WITH_HIP="${USE_HIP}"
 WALBERLA_BUILD_WITH_OPENMP=${USE_OMP}
+CMAKE_CUDA_ARCHITECTURES="${CUDA_COMPUTE_CAP}"
 
 echo "Building walberla in $(pwd)"
 
@@ -76,6 +68,7 @@ cmake ${FULL_PATH} \
     -DWALBERLA_OPTIMIZE_FOR_LOCALHOST=$WALBERLA_OPTIMIZE_FOR_LOCALHOST \
     -DWALBERLA_BUILD_WITH_FASTMATH=$WALBERLA_BUILD_WITH_FASTMATH \
     -DWALBERLA_BUILD_WITH_LTO=$WALBERLA_BUILD_WITH_LTO \
+    -DCMAKE_CUDA_ARCHITECTURES="${CMAKE_CUDA_ARCHITECTURES}" \
     -DWALBERLA_LOGLEVEL=$WALBERLA_LOGLEVEL
 
 cmake . -LA