gpu target for elpa and cray math-mode for cosma

This commit is contained in:
Fabian Ducry 2021-10-19 17:45:41 +02:00 committed by Ole Schütt
parent e6ceced390
commit a8e514d031
2 changed files with 23 additions and 34 deletions

View file

@ -52,24 +52,25 @@ case "$with_cosma" in
cd build-cpu
case "$MATH_MODE" in
mkl)
cmake \
-DCMAKE_INSTALL_PREFIX="${pkg_install_dir}" \
-DCMAKE_INSTALL_LIBDIR=lib \
-DCOSMA_BLAS=MKL \
-DCOSMA_SCALAPACK=MKL \
-DCOSMA_WITH_TESTS=NO \
-DCOSMA_WITH_APPS=NO .. > cmake.log 2>&1
cosma_blas='MKL'
cosma_sl='MKL'
;;
cray)
cosma_blas='CRAY_LIBSCI'
cosma_sl='CRAY_LIBSCI'
;;
*)
cmake \
-DCMAKE_INSTALL_PREFIX="${pkg_install_dir}" \
-DCMAKE_INSTALL_LIBDIR=lib \
-DCOSMA_BLAS=OPENBLAS \
-DCOSMA_SCALAPACK=CUSTOM \
-DCOSMA_WITH_TESTS=NO \
-DCOSMA_WITH_APPS=NO .. > cmake.log 2>&1
cosma_blas='OPENBLAS'
cosma_sl='CUSTOM'
;;
esac
cmake \
-DCMAKE_INSTALL_PREFIX="${pkg_install_dir}" \
-DCMAKE_INSTALL_LIBDIR=lib \
-DCOSMA_BLAS=${cosma_blas} \
-DCOSMA_SCALAPACK=${cosma_sl} \
-DCOSMA_WITH_TESTS=NO \
-DCOSMA_WITH_APPS=NO .. > cmake.log 2>&1
make -j $(get_nprocs) > make.log 2>&1
make -j $(get_nprocs) install > install.log 2>&1
cd ..
@ -79,26 +80,13 @@ case "$with_cosma" in
[ -d build-cuda ] && rm -rf "build-cuda"
mkdir build-cuda
cd build-cuda
case "$MATH_MODE" in
mkl)
cmake \
-DCMAKE_INSTALL_PREFIX="${pkg_install_dir}-cuda" \
-DCMAKE_INSTALL_LIBDIR=lib \
-DCOSMA_BLAS=CUDA \
-DCOSMA_SCALAPACK=MKL \
-DCOSMA_WITH_TESTS=NO \
-DCOSMA_WITH_APPS=NO .. > cmake.log 2>&1
;;
*)
cmake \
-DCMAKE_INSTALL_PREFIX="${pkg_install_dir}-cuda" \
-DCMAKE_INSTALL_LIBDIR=lib \
-DCOSMA_BLAS=CUDA \
-DCOSMA_SCALAPACK=CUSTOM \
-DCOSMA_WITH_TESTS=NO \
-DCOSMA_WITH_APPS=NO .. > cmake.log 2>&1
;;
esac
cmake \
-DCMAKE_INSTALL_PREFIX="${pkg_install_dir}-cuda" \
-DCMAKE_INSTALL_LIBDIR=lib \
-DCOSMA_BLAS=CUDA \
-DCOSMA_SCALAPACK=${cosma_sl} \
-DCOSMA_WITH_TESTS=NO \
-DCOSMA_WITH_APPS=NO .. > cmake.log 2>&1
make -j $(get_nprocs) > make.log 2>&1
make -j $(get_nprocs) install > install.log 2>&1
cd ..

View file

@ -115,6 +115,7 @@ case "$with_elpa" in
${config_flags} \
--enable-nvidia-gpu=$([ "$TARGET" == "nvidia" ] && echo "yes" || echo "no") \
--with-cuda-path=${CUDA_PATH} \
--with-NVIDIA-GPU-compute-capability=$([ "$TARGET" == "nvidia" ] && echo "sm_$ARCH_NUM" || echo "sm_35") \
OMPI_MCA_plm_rsh_agent=/bin/false \
FC=${MPIFC} \
CC=${MPICC} \